mirror of
https://github.com/Maks1mS/altlinux-autorepacked.git
synced 2024-12-23 14:42:59 +03:00
124 lines
3.3 KiB
Python
124 lines
3.3 KiB
Python
import datetime
|
|
import json
|
|
import os
|
|
import importlib
|
|
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import BackgroundTasks, FastAPI
|
|
import uvicorn
|
|
from fastapi_utils.tasks import repeat_every
|
|
|
|
from autorepacked import utils
|
|
from autorepacked.common_providers.base_provider import BaseProvider
|
|
from autorepacked.config import Config
|
|
|
|
|
|
def create_repo(config: Config):
|
|
utils.epm([
|
|
'repo',
|
|
'create',
|
|
config.get('repo_path')
|
|
])
|
|
|
|
|
|
def update_cache():
|
|
utils.epm(['update'])
|
|
|
|
def update_epm():
|
|
utils.epm(['ei'])
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
await update_repeat()
|
|
yield
|
|
|
|
|
|
config = Config()
|
|
app = FastAPI(lifespan=lifespan, root_path=config.get('root_path'))
|
|
|
|
update_task_started = False
|
|
|
|
|
|
def update():
|
|
json_file_path = os.path.join(config.get('repo_path'), 'data.json')
|
|
if not os.path.exists(json_file_path):
|
|
with open(json_file_path, 'w') as data_file:
|
|
json.dump({'versions': {}}, data_file, indent=4)
|
|
with open(json_file_path, 'r') as data_file:
|
|
repo_data = json.load(data_file)
|
|
|
|
global update_task_started
|
|
|
|
if 'last_version_check' in repo_data:
|
|
last_version_check = datetime.datetime.fromisoformat(repo_data['last_version_check'])
|
|
|
|
if datetime.datetime.now() - datetime.timedelta(minutes=5) < last_version_check:
|
|
update_task_started = False
|
|
return
|
|
|
|
update_cache()
|
|
update_epm()
|
|
|
|
providers_path = os.path.join(os.getcwd(), 'autorepacked/providers')
|
|
modules = os.listdir(providers_path)
|
|
|
|
versions = repo_data['versions']
|
|
need_update_index = False
|
|
for module_name in modules:
|
|
module = importlib.import_module(f'providers.{module_name}')
|
|
if hasattr(module, 'get_provider'):
|
|
provider = module.get_provider(config) # type: BaseProvider
|
|
if not provider.disabled:
|
|
name = provider.get_name()
|
|
print(f"request version of {name}")
|
|
version = provider.get_version()
|
|
if name not in versions or versions[name] != version:
|
|
print(f'{name} has new version {version}')
|
|
versions[name] = version
|
|
provider.download()
|
|
need_update_index = True
|
|
|
|
repo_data['versions'] = versions
|
|
|
|
current_date = datetime.datetime.utcnow()
|
|
|
|
if need_update_index:
|
|
utils.epm(
|
|
args=[
|
|
'repo',
|
|
'index',
|
|
config.get('repo_path'),
|
|
]
|
|
)
|
|
repo_data['last_index_update'] = current_date.isoformat()
|
|
|
|
repo_data['last_version_check'] = current_date.isoformat()
|
|
|
|
with open(json_file_path, 'w') as data_file:
|
|
json.dump(repo_data, data_file, indent=4)
|
|
|
|
update_task_started = False
|
|
|
|
|
|
@app.post("/update")
|
|
def update_method(background_tasks: BackgroundTasks):
|
|
global update_task_started
|
|
if not update_task_started:
|
|
update_task_started = True
|
|
background_tasks.add_task(update)
|
|
return {'status': 'OK'}
|
|
|
|
|
|
@repeat_every(seconds=60 * 60 * 4)
|
|
def update_repeat():
|
|
global update_task_started
|
|
if not update_task_started:
|
|
update_task_started = True
|
|
update()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
create_repo(config)
|
|
uvicorn.run(app, host="0.0.0.0", port=8000)
|