You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
2.6 KiB
60 lines
2.6 KiB
from get_general_settings import *
|
|
|
|
import os
|
|
import pygit2
|
|
import logging
|
|
import sqlite3
|
|
|
|
current_working_directory = os.path.dirname(__file__)
|
|
repository_path = pygit2.discover_repository(current_working_directory)
|
|
local_repo = pygit2.Repository(repository_path)
|
|
|
|
def check_and_apply_update(repo=local_repo, remote_name='origin'):
|
|
repo.config['remote.origin.fetch'] = '+refs/heads/*:refs/remotes/origin/*'
|
|
repo.remotes[remote_name].fetch()
|
|
repo.config['user.name'] = 'Bazarr user'
|
|
repo.config['user.email'] ='bazarr@fakeuser.com'
|
|
|
|
for remote in repo.remotes:
|
|
if remote.name == remote_name:
|
|
remote.fetch()
|
|
remote_id = repo.lookup_reference('refs/remotes/origin/' + str(branch)).target
|
|
merge_result, _ = repo.merge_analysis(remote_id)
|
|
# Up to date, do nothing
|
|
if merge_result & pygit2.GIT_MERGE_ANALYSIS_UP_TO_DATE:
|
|
logging.info('No new version of Bazarr available.')
|
|
pass
|
|
# We can just fastforward
|
|
elif merge_result & pygit2.GIT_MERGE_ANALYSIS_FASTFORWARD:
|
|
repo.checkout_tree(repo.get(remote_id))
|
|
master_ref = repo.lookup_reference('refs/remotes/origin/' + str(branch))
|
|
master_ref.set_target(remote_id)
|
|
repo.head.set_target(remote_id)
|
|
logging.info('Bazarr updated to latest version and need to be restarted.')
|
|
updated()
|
|
# We can just do it normally
|
|
elif merge_result & pygit2.GIT_MERGE_ANALYSIS_NORMAL:
|
|
repo.merge(remote_id)
|
|
print repo.index.conflicts
|
|
|
|
assert repo.index.conflicts is None, 'Conflicts, ahhhh!'
|
|
user = repo.default_signature
|
|
tree = repo.index.write_tree()
|
|
commit = repo.create_commit('HEAD',
|
|
user,
|
|
user,
|
|
'Merge!',
|
|
tree,
|
|
[repo.head.target, remote_id])
|
|
repo.state_cleanup()
|
|
logging.error('Conflict detected when trying to update.')
|
|
# We can't do it
|
|
else:
|
|
logging.error('Bazarr cannot be updated: Unknown merge analysis result')
|
|
|
|
def updated():
|
|
conn = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'), timeout=30)
|
|
c = conn.cursor()
|
|
c.execute("UPDATE table_settings_general SET updated = 1")
|
|
conn.commit()
|
|
c.close() |