crawl_xiaohua/main.py

#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@Contact :   liuyuqi.gov@msn.cn
@Time    :   2022/05/23 14:33:19
@License :   Copyright © 2017-2022 liuyuqi. All Rights Reserved.
@Desc    :   main
'''
import crawl_xiaohua
import argparse

parser = argparse.ArgumentParser(description='crawl_xiaohua')
parser.add_argument('command', type=str, help='server, run ?')
parser.add_argument('--config', type=str, help='config file')
parser.add_argument('--extractor', type=str, help='extractor name')
parser.add_argument('--cmd', type=str, help='shell command')

if __name__ == '__main__':
    try:
        args = parser.parse_args()
        if args.command == 'server':
            crawl_xiaohua.server(args.config)
            print('server')
        elif args.command == 'run':
            crawl_xiaohua.run(args.extractor, args.cmd)
        else:
            print('unknown command')
            parser.print_help()
    except Exception as e:
        print(e)
        parser.print_help()
    finally:
        pass