proxy_pool/proxyPool.py

# -*- coding: utf-8 -*-
"""
-------------------------------------------------
   File Name：     proxy_pool
   Description :   proxy pool 启动入口
   Author :        JHao
   date：          2020/6/19
-------------------------------------------------
   Change Activity:
                   2020/6/19:
-------------------------------------------------
"""
import os

__author__ = 'JHao'

import click

from setting import BANNER, VERSION

CONTEXT_SETTINGS = dict(help_option_names=['-h', '--help'])


@click.group(context_settings=CONTEXT_SETTINGS)
@click.version_option(version=VERSION)
def cli():
    """ProxyPool cli工具"""


@cli.command(name="schedule")
def schedule():
    """ 启动调度程序 """
    click.echo(BANNER)
    click.echo("VERSION: %s\n" % VERSION)
    from helper.scheduler import runScheduler
    runScheduler()


@cli.command(name="server")
def server():
    """ 启动api服务 """
    click.echo(BANNER)
    click.echo("VERSION: %s\n" % VERSION)
    from api.proxyApi import runFlask
    runFlask()


if __name__ == '__main__':
    os.environ['TZ'] = 'Asia/Shanghai'
    cli()