71 lines
1.7 KiB
Python
71 lines
1.7 KiB
Python
from pydantic import AnyHttpUrl
|
|
from pathlib import Path
|
|
from git import Repo
|
|
import logging
|
|
import shutil
|
|
import os
|
|
from configparser import ConfigParser
|
|
from runner_script import RunnerScript
|
|
|
|
|
|
# load default values
|
|
config = ConfigParser()
|
|
config.read(Path(__file__).parent / 'defaults.ini')
|
|
REPO_DIR = config.get('main', 'repo_dir')
|
|
|
|
|
|
def clone_repository(
|
|
repo_url: AnyHttpUrl,
|
|
repo_dir: str = REPO_DIR
|
|
) -> Path:
|
|
# extract repo name
|
|
repo_name = repo_url.split('/')[-1].replace('.git', '')
|
|
dst_dir = Path(repo_dir) / repo_name
|
|
# prepare repo download dir
|
|
if dst_dir.exists():
|
|
shutil.rmtree(dst_dir)
|
|
dst_dir.mkdir()
|
|
# git clone repo
|
|
Repo.clone_from(url=repo_url, to_path=dst_dir)
|
|
logging.debug('finished')
|
|
return dst_dir
|
|
|
|
|
|
def execute_script(
|
|
local_repo_dir: Path
|
|
) -> None:
|
|
# locate the script
|
|
script_path = (
|
|
local_repo_dir /
|
|
'.gitea' /
|
|
'gpu_runner' /
|
|
'script.yml'
|
|
)
|
|
# determine original working dir
|
|
org_dir = os.getcwd()
|
|
# initialise command list
|
|
cmd_list = [
|
|
f'cd {local_repo_dir.resolve()}',
|
|
'virtualenv -p python3.10 venv',
|
|
'source venv/bin/activate',
|
|
]
|
|
# add in runner script
|
|
rs = RunnerScript.from_file(path=script_path)
|
|
for cmd in rs.script:
|
|
cmd_list.append(cmd)
|
|
# append cleanup commands
|
|
cmd_list.append('deactivate')
|
|
cmd_list.append(f'cd {org_dir}')
|
|
cmd_list.append(f'rm -rf {local_repo_dir.resolve()}')
|
|
# build command string
|
|
cmd_str = ' && '.join(cmd_list)
|
|
os.system(cmd_str)
|
|
logging.debug('finished')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
local_repo_dir = Path(__file__).parent.parent / 'repo' / 'gpu_runner'
|
|
execute_script(
|
|
local_repo_dir=local_repo_dir
|
|
)
|