jasima/majuna
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cli/automate: fix up errors found with mypy

Browse source
This commit is contained in:
Iain Learmonth 2022-05-16 09:24:37 +01:00
parent 55a0b19c8c
commit ccf0ce6a06
3 changed files with 55 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

27
app/cli/automate.py
View file

@ -3,6 +3,7 @@ import datetime
import json
import logging
from traceback import TracebackException
from typing import Type, TYPE_CHECKING, Any
from app import app
from app.extensions import db
@ -28,6 +29,11 @@ from app.terraform.proxy.azure_cdn import ProxyAzureCdnAutomation
from app.terraform.proxy.cloudfront import ProxyCloudfrontAutomation
if TYPE_CHECKING:
_SubparserType = argparse._SubParsersAction[argparse.ArgumentParser]
else:
_SubparserType = Any
jobs = {
x.short_name: x
for x in [
@ -52,17 +58,18 @@ jobs = {
}
def run_all(**kwargs):
def run_all(**kwargs: bool) -> None:
for job in jobs.values():
run_job(job, **kwargs)
def run_job(job: BaseAutomation, *, force: bool = False, ignore_schedule: bool = False):
automation = Automation.query.filter(Automation.short_name == job.short_name).first()
def run_job(job_cls: Type[BaseAutomation], *,
force: bool = False, ignore_schedule: bool = False) -> None:
automation = Automation.query.filter(Automation.short_name == job_cls.short_name).first()
if automation is None:
automation = Automation()
automation.short_name = job.short_name
automation.description = job.description
automation.short_name = job_cls.short_name
automation.description = job_cls.description
automation.enabled = False
automation.next_is_full = False
automation.added = datetime.datetime.utcnow()
@ -78,11 +85,11 @@ def run_job(job: BaseAutomation, *, force: bool = False, ignore_schedule: bool =
logging.warning("Not time to run this job yet")
return
if not automation.enabled and not force:
logging.warning(f"job {job.short_name} is disabled and --force not specified")
logging.warning(f"job {job_cls.short_name} is disabled and --force not specified")
return
automation.state = AutomationState.RUNNING
db.session.commit()
job = job()
job: BaseAutomation = job_cls()
try:
success, logs = job.automate()
except Exception as e:
@ -114,7 +121,7 @@ def run_job(job: BaseAutomation, *, force: bool = False, ignore_schedule: bool =
class AutomateCliHandler:
@classmethod
def add_subparser_to(cls, subparsers: argparse._SubParsersAction) -> None:
def add_subparser_to(cls, subparsers: _SubparserType) -> None:
parser = subparsers.add_parser("automate", help="automation operations")
parser.add_argument("-a", "--all", dest="all", help="run all automation jobs", action="store_true")
parser.add_argument("-j", "--job", dest="job", choices=sorted(jobs.keys()),
@ -123,10 +130,10 @@ class AutomateCliHandler:
parser.add_argument("--ignore-schedule", help="run job even if it's not time yet", action="store_true")
parser.set_defaults(cls=cls)
def __init__(self, args):
def __init__(self, args: argparse.Namespace) -> None:
self.args = args
def run(self):
def run(self) -> None:
with app.app_context():
if self.args.job:
run_job(jobs[self.args.job], force=self.args.force, ignore_schedule=self.args.ignore_schedule)