forked from lllyasviel/Fooocus
-
Notifications
You must be signed in to change notification settings - Fork 20
/
entry_with_update.py
49 lines (38 loc) · 1.41 KB
/
entry_with_update.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import os
import sys
from pathlib import Path
root = Path(__file__).resolve().parent
sys.path.append(str(root))
os.chdir(root)
bupdated = False
try:
import pygit2
pygit2.option(pygit2.GIT_OPT_SET_OWNER_VALIDATION, 0)
repo_path = Path(__file__).resolve().parent
repo = pygit2.Repository(str(repo_path))
branch_name = repo.head.shorthand
remote_name = "origin"
remote = repo.remotes[remote_name]
remote.fetch()
local_branch_ref = f"refs/heads/{branch_name}"
local_branch = repo.lookup_reference(local_branch_ref)
remote_reference = f"refs/remotes/{remote_name}/{branch_name}"
remote_commit = repo.revparse_single(remote_reference)
merge_result, _ = repo.merge_analysis(remote_commit.id)
if merge_result & pygit2.GIT_MERGE_ANALYSIS_UP_TO_DATE:
print("You have the latest version")
elif merge_result & pygit2.GIT_MERGE_ANALYSIS_FASTFORWARD:
local_branch.set_target(remote_commit.id)
repo.head.set_target(remote_commit.id)
repo.checkout_tree(repo.get(remote_commit.id))
repo.reset(local_branch.target, pygit2.GIT_RESET_HARD)
print("Updating Files")
bupdated = True
elif merge_result & pygit2.GIT_MERGE_ANALYSIS_NORMAL:
print("Update failed, Did you modify any files?")
except Exception as e:
print("Update failed...")
print(str(e))
if bupdated:
print("Update succeeded!!")
from launch import *