main.py

# -*- coding: utf-8 -*-
# Time       : 2022/2/15 17:40
# Author     : QIN2DIM
# Github     : https://github.com/QIN2DIM
# Description: 🚀 Yo Challenger!
import typing
import warnings

from fire import Fire

from examples import demo_selenium, demo_challenge, demo_install, demo_classify
from examples.motion import app, motion
from examples.settings import config

demo_install.do()

warnings.filterwarnings("ignore", category=DeprecationWarning)


class Scaffold:
    """System scaffolding Top-level interface commands"""

    CHALLENGE_LANGUAGE = "en"

    def __init__(self, lang: typing.Optional[str] = None):
        if lang is not None:
            Scaffold.CHALLENGE_LANGUAGE = lang

        self.challenge = self.demo
        self.run = self.demo

    @staticmethod
    def install(model: typing.Optional[str] = None, upgrade: typing.Optional[bool] = False):
        """Download Project Dependencies and upgrade all pluggable ONNX model"""
        demo_install.do(yolo_onnx_prefix=model, upgrade=upgrade)

    @staticmethod
    def test():
        """Test the Challenger drive for fitment"""
        demo_challenge.test()

    @staticmethod
    def tracker():
        app.run(debug=True, access_log=False)

    @staticmethod
    def motion():
        motion.train_motion("http://127.0.0.1:8000", config.dir_database)

    @staticmethod
    def demo(
        silence: typing.Optional[bool] = False,
        model: typing.Optional[str] = None,
        target: typing.Optional[str] = None,
        sitekey: typing.Optional[str] = None,
        screenshot: typing.Optional[bool] = False,
        repeat: typing.Optional[int] = 5,
        slowdown: typing.Optional[bool] = True,
    ):
        """
        Dueling with hCAPTCHA challenge using YOLOv5.

        Usage: python main.py demo [flags]
        ——————————————————————————————————————————————————————————————————
        or: python main.py demo --model=yolov5n6
        or: python main.py demo --target=discord
        or: python main.py demo --lang=en
        or: python main.py demo --sitekey=[UUID]
        or: pythoon main.py demo --slowdown=False
        ——————————————————————————————————————————————————————————————————

        :param slowdown: Add a short sleep so that the user can see the prediction results of the model
        :param repeat: Default 5. Number of times to repeat the presentation.
        :param screenshot: Default False. Save screenshot of the challenge result.
            PATH: database/temp_cache/captcha_screenshot/
            FILENAME: ChallengeLabelName.png
        :param sitekey: Default None. customize the challenge theme via sitekey
        :param silence: Default False. Whether to startup headless browser.
        :param model: Default "yolov5s6". Select the YOLO model to handle specific challenges.
            within [yolov5n6 yolov5s6 yolov5m6 yolov6n yolov6t yolov6s]
        :param target: Default None. Designate `Challenge Source`. See the global value SITE_KEYS.
        :return:
        """
        # Generate challenge topics
        if config.SITE_KEYS.get(target):
            sample_site = config.HCAPTCHA_DEMO_API.format(config.SITE_KEYS[target])
        else:
            sample_site = config.HCAPTCHA_DEMO_SITES[0]
        if sitekey is not None:
            sample_site = config.HCAPTCHA_DEMO_API.format(sitekey.strip())

        demo_challenge.run(
            sample_site,
            lang=Scaffold.CHALLENGE_LANGUAGE,
            silence=silence,
            onnx_prefix=model,
            screenshot=screenshot,
            repeat=repeat,
            slowdown=slowdown,
        )

    @staticmethod
    def demo_bytedance():
        """
        signup hcaptcha dashboard

        Usage: python main.py demo-bytedance
        :return:
        """
        demo_selenium.bytedance()

    @staticmethod
    def demo_classify():
        """
        BinaryClassification Task

        Usage: python main.py demo-classify
        :return:
        """
        demo_classify.bytedance()


if __name__ == "__main__":
    Fire(Scaffold)