45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
import json
|
|
import sys
|
|
from multiprocessing import Pool
|
|
|
|
from importlib import import_module
|
|
|
|
from db import *
|
|
|
|
|
|
def get_game() -> list:
|
|
local_db = get_local_db('admin_game')
|
|
games = list(local_db['game'].find())
|
|
return games
|
|
|
|
|
|
def run_task(kwargs):
|
|
module_name = kwargs.get('task_name')
|
|
class_name = ''.join([s.capitalize() for s in module_name.split('_')])
|
|
module = import_module(f'.{module_name}', package='task')
|
|
c_obj = getattr(module, class_name)
|
|
obj = c_obj(**kwargs)
|
|
obj.run()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# eg: summary_func 0 0
|
|
# eg: first_recharge 0 0
|
|
task_name, st, et = sys.argv[1:]
|
|
st, et = int(st), int(et)
|
|
game_list = get_game()
|
|
with open('config.json', 'r', encoding='utf8') as f:
|
|
task_conf = json.load(f)
|
|
params = []
|
|
for item in game_list:
|
|
p = {'game_name': item['id_name'],
|
|
'timezone': item.get('timezone', 'Asia/Shanghai'),
|
|
'st': st,
|
|
'et': et
|
|
}
|
|
p.update(task_conf[task_name])
|
|
params.append(p)
|
|
|
|
with Pool(len(game_list)) as p:
|
|
p.map(run_task, params)
|