Archived
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -- coding:utf-8 --
|
||
|
|
# @Author: markushammered@gmail.com
|
||
|
|
# @Development Tool: Pycharm
|
||
|
|
# @Create Time: 2022/4/29
|
||
|
|
# @File Name: update.py
|
||
|
|
|
||
|
|
|
||
|
|
import os
|
||
|
|
import base64
|
||
|
|
import logging
|
||
|
|
import requests
|
||
|
|
from ..config import VersionConfig
|
||
|
|
from typing import Dict, List
|
||
|
|
|
||
|
|
|
||
|
|
class SelfUpdateVersion:
|
||
|
|
"""更新本体 -> 更新单个文件"""
|
||
|
|
|
||
|
|
def __init__(self, logger: logging.Logger):
|
||
|
|
self.api = 'https://api.github.com/repos/MarkusJoe/RequestCounter/tags'
|
||
|
|
self.headers = {'Authorization': 'token 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'}
|
||
|
|
self.resp = requests.get(self.api, headers=self.headers).json()
|
||
|
|
self.logger = logger
|
||
|
|
|
||
|
|
def commit_details(self):
|
||
|
|
"""
|
||
|
|
获取提交细节
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
commit_url = self.resp[0]['commit']['url']
|
||
|
|
res = requests.get(commit_url, headers=self.headers).json()
|
||
|
|
files = res['files']
|
||
|
|
modified = []
|
||
|
|
deleted = []
|
||
|
|
|
||
|
|
for f in files:
|
||
|
|
file_path = f['filename']
|
||
|
|
base64_content = requests.get(f['contents_url'], headers=self.headers).json()['content'] # 直接获取修改文件的base64
|
||
|
|
if f['status'] in ['modified', 'added']:
|
||
|
|
modified.append({'filename': f'./{file_path}', 'content': base64.b64decode(base64_content)})
|
||
|
|
else:
|
||
|
|
deleted.append(f'./{file_path}')
|
||
|
|
|
||
|
|
self.apply_change(modified, deleted)
|
||
|
|
|
||
|
|
def apply_change(self, modified: List[Dict], deleted: List):
|
||
|
|
"""
|
||
|
|
应用已经更改的文件
|
||
|
|
:param modified:
|
||
|
|
:param deleted:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
for d in deleted:
|
||
|
|
os.remove(d)
|
||
|
|
self.logger.info(f'删除了: {d}')
|
||
|
|
|
||
|
|
for m in modified:
|
||
|
|
with open(m['filename'], 'wb') as modify:
|
||
|
|
modify.write(m['content'])
|
||
|
|
self.logger.info(f'修改了: {m["filename"]}')
|
||
|
|
|
||
|
|
with open('./app/config.py', 'wb') as update_config:
|
||
|
|
content = requests.get('https://api.github.com/repos/MarkusJoe/RequestCounter/contents/app/config.py',
|
||
|
|
headers=self.headers).json()['content']
|
||
|
|
update_config.write(base64.b64decode(content))
|
||
|
|
|
||
|
|
exit()
|
||
|
|
|
||
|
|
def check(self):
|
||
|
|
current = VersionConfig.version[0]
|
||
|
|
newest = self.resp[0]['name']
|
||
|
|
if current != newest:
|
||
|
|
self.logger.warning(f'当前版本已经落后于最新版本. 当前: {current} -> 远程: {newest}')
|
||
|
|
self.commit_details()
|
||
|
|
return False
|
||
|
|
else:
|
||
|
|
self.logger.info('当前版本为最新版本')
|
||
|
|
return True
|