DailyCheck/DailyCheck.py at master · toneoesa/DailyCheck · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import random
from random import randint
import argparse
from time import asctime, sleep
from check import check
from LogGroup import LogGroup

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Auto Daily Check Script')
    parser.add_argument('-n', '--now', default=False, action='store_true', help='Do not delay')
    parser.add_argument('-d', '--dry', default=False, action='store_true', help='Dry run, no real check')
    parser.add_argument('-s', '--silent', default=False, action='store_true', help='Do not report result to DingDing')
    parser.add_argument('-f', '--file', default='idfile.txt', action='store',
                        help='A file with id(s), password(s) (and ding token, secret)')
    parser.add_argument('-i', '--id', default=None, action='store', help='Specify a id')
    parser.add_argument('-p', '--pwd', default=None, action='store', help='Specify a password')
    parser.add_argument('-l', '--logfile', default='log.txt', action='store', help='File to append log')
    parser.add_argument('-S', '--sleep', default=600, type=int, action='store', help='Sleep seconds before work')
    parser.add_argument('-t', '--time', default=600, type=int, action='store',
                        help='Choose time points randomly in during this time')
    parser.add_argument('--shuffle', default=False, action='store_true', help='Shuffle the order of users')
    args = parser.parse_args().__dict__

    access_token = ''
    secret = ''
    ids = []
    if args['file'] is not None:
        try:
            with open(args['file'], 'r', encoding='utf-8') as id_file:
                for line in id_file:
                    if len(line.replace('\n', '')) == 0 or line[0] == '#':
                        continue
                    if line[0] == '@':
                        access_token, secret = line[1:].split()[:2]
                    else:
                        info = line.split()[:3]
                        ids.append(info)
        except:
            pass

    if args['id'] is not None and args['pwd'] is not None:
        ids.append([args['id'], args['pwd'], 'TempUser'])

    if len(ids) == 0:
        print('Specify a file or a pair of id and pwd')
        exit(2)

    time_max_delay = int(args['time'] / len(ids)) + 1
    log_group = LogGroup(print='print', file='write_now',
                         ding='ding_together' if not args['silent'] and access_token and secret else 'nul')
    log_group.set_path('file', args['logfile'])
    log_group.set_ding(access_token, secret)
    log_group.write('From Daily Check:', 'ding')
    log_group.write(f'{asctime()}')

    if args['shuffle']:
        random.shuffle(ids)

    if not args['now']:
        sleep(args['sleep'])

    for id, pwd, name in ids:
        if not args['now']:
            sleep(randint(1, time_max_delay))
        log_group.write(f'  {asctime()}')
        log_group.write(f'    {id} {name}')
        if not args['dry']:
            msg = check(id, pwd)
            log_group.write(f"    {msg['Info']}")
        else:
            log_group.write(f'    Dry Run')

    log_group.write('Check all done')
    log_group.write('')
    log_group.execute()