-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
executable file
·89 lines (80 loc) · 2.52 KB
/
main.py
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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
#!/usr/bin/python3
# coding: utf8
from server import *
from bot import Bot
import json
import logging
import logging.config
import os
import os.path
def parse():
import argparse
parser = argparse.ArgumentParser(description='Starts a bot')
parser.add_argument("--purge", help="purges stuff", action="store_true")
parser.add_argument("--webhook", help="uses webhook", type=str)
parser.add_argument("--install", help="install with tocken", type=str, metavar='TOCKEN')
args = parser.parse_args()
return args.purge, args.webhook, args.install
def main():
DIR = os.path.join(os.path.dirname(__file__), "botTest")
TOCKEN_PATH = os.path.join(DIR, "token")
LOGGING_PATH = os.path.join(os.path.join(os.path.dirname(__file__), "logs"), "config.json")
PRIVATE_KEY_PATH = os.path.join(DIR, "private.key")
PUBLIC_KEY_PATH = os.path.join(DIR, "public.pem")
purge, webhook_url, install = parse()
if install != None:
if not os.path.exists(DIR):
os.mkdir(DIR)
with open(TOCKEN_PATH, 'w') as f:
f.write(install+"\n")
print("End installing")
return
if os.path.exists(LOGGING_PATH):
with open(LOGGING_PATH, 'rt') as f:
config = json.load(f)
logging.config.dictConfig(config)
else:
logging.basicConfig(level=logging.DEBUG)
with open(TOCKEN_PATH, 'r') as f:
token = f.readline()[:-1]
bot = Bot(token, DIR)
if purge:
bot.purge()
return
server = None
if webhook_url:
if not os.path.exists(PUBLIC_KEY_PATH):
print("No public key, please run ssl.sh before !")
return
if not os.path.exists(PRIVATE_KEY_PATH):
print("No private key, please run ssl.sh before !")
return
server = WebHookServer(bot, PUBLIC_KEY_PATH, PRIVATE_KEY_PATH, webhook_url)
else:
server = PollingServer(bot)
print("### Start bot...")
bot.start()
print("### Bot started")
print("### Start server...")
server.start()
print("### Server started")
try:
while True:
time.sleep(60*60)
except KeyboardInterrupt:
pass
print("### Stop server...")
server.stop()
print("### Server stopped")
print("### Stop bot...")
bot.stop()
print("### Bot stopped")
print("### Join server...")
server.join()
print("### Server joined")
print("### Join bot...")
bot.join()
print("### Bot joined")
print("### End of main")
if __name__ == "__main__":
main()