File size: 2,811 Bytes
8a469fd |
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 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
import os
import re
import requests
import subprocess
import hashlib
def get_git_commit_version():
extension_dir = os.path.dirname(os.path.abspath(__file__)) + '/../../'
extension_dir = os.path.normpath(extension_dir)
git_path = os.path.join(extension_dir, '.git')
if os.path.exists(git_path):
try:
git = os.environ.get('GIT', "git")
if not git:
git = "git"
cmd = [git, 'rev-parse', 'HEAD']
commit_version = subprocess.check_output(cmd, cwd=extension_dir).decode('utf-8').strip()
if re.match(r'^[0-9a-f]{40}$', commit_version):
return commit_version
except Exception as e:
pass
try:
ref_path = os.path.join(git_path, 'refs', 'heads', 'main')
with open(ref_path, 'r') as f:
commit_version = f.read().strip()
if re.match(r'^[0-9a-f]{40}$', commit_version):
return commit_version
except Exception as e:
pass
return ''
def _handle_versions(response, filter_update_readme=False):
try:
if response.status_code != 200:
return []
result = response.json()
if not result:
return []
versions = []
for item in result:
message = item['commit']['message']
is_update_readme = False
if message.lower().strip() == 'update readme.md':
if filter_update_readme:
continue
is_update_readme = True
versions.append({
'version': item['sha'],
'message': message,
'date': item['commit']['committer']['date'],
'is_update_readme': is_update_readme
})
return versions
except Exception as e:
return []
def get_git_remote_versions(page=1, per_page=100, filter_update_readme=False):
api_urls = [
'https://api.github.com/repos/physton/sd-webui-prompt-all-in-one/commits',
'https://gitee.com/api/v5/repos/physton/sd-webui-prompt-all-in-one/commits'
]
for api_url in api_urls:
try:
api_url += f'?page={page}&per_page={per_page}'
key = hashlib.md5(api_url.encode('utf-8')).hexdigest()
response = requests.get(api_url)
versions = _handle_versions(response, filter_update_readme)
return versions
except Exception as e:
pass
return []
def get_latest_version():
current_version = get_git_commit_version()
# if not current_version:
# return current_version
versions = get_git_remote_versions(1, 10, False)
if len(versions) < 1:
return current_version
return versions[0]['version']
|