-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.py
95 lines (73 loc) · 2.64 KB
/
run.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
90
91
92
93
94
95
import os
import argparse
def main():
arg_parser = argparse.ArgumentParser(description='bsc Python code obfuscation local demo server')
arg_parser.add_argument(
'-p', '--port',
default=5000, type=int,
help='Port to run'
)
arg_parser.add_argument(
'-d', '--debug',
default=False, action='store_true',
help='Enable run in debug mode'
)
arg_parser.add_argument(
'--db_url',
default='mongodb://localhost:27017/', type=str,
help='url mongod is listening on'
)
arg_parser.add_argument(
'--db_name',
default='python-code-obfuscation', type=str,
help='data base name'
)
arg_parser.add_argument(
'--db_collection',
default='source-codes', type=str,
help='data base collection'
)
arg_parser.add_argument(
'--text_to_image_service_url',
default='localhost', type=str,
help='text to image service url'
)
arg_parser.add_argument(
'--text_to_image_service_port',
default=8080, type=int,
help='text to image service port'
)
arg_parser.add_argument(
'--max_tmp_files',
default=1000, type=int,
help='maximum number of simultaneously stored files in RAM'
)
arg_parser.add_argument(
'--max_tmp_files_age',
default=86400, type=int,
help='maximum age in seconds of temporary files stored in RAM'
)
arg_parser.add_argument(
'--log_dir',
default='.', type=str,
help='logging directory of application'
)
arg_parser.add_argument(
'--log_level',
default=20, type=int,
help='logging level of application: DEBUG=10, INFO=20, WARNING=30, ERROR=40, CRITICAL=50'
)
args = arg_parser.parse_args()
os.environ['PYTHON_CODE_OBFUSCATION_LOG_DIR'] = args.log_dir
os.environ['PYTHON_CODE_OBFUSCATION_LOG_LEVEL'] = str(args.log_level)
os.environ['PYTHON_CODE_OBFUSCATION_MONGODB_URL'] = args.db_url
os.environ['PYTHON_CODE_OBFUSCATION_MONGODB_DB_NAME'] = args.db_name
os.environ['PYTHON_CODE_OBFUSCATION_MONGODB_DB_COLLECTION'] = args.db_collection
os.environ['PYTHON_CODE_OBFUSCATION_TEXT_TO_IMAGE_SERVICE_URL'] = args.text_to_image_service_url
os.environ['PYTHON_CODE_OBFUSCATION_TEXT_TO_IMAGE_SERVICE_PORT'] = str(args.text_to_image_service_port)
os.environ['PYTHON_CODE_OBFUSCATION_MAX_TMP_FILES'] = str(args.max_tmp_files)
os.environ['PYTHON_CODE_OBFUSCATION_MAX_TMP_FILES_AGE'] = str(args.max_tmp_files_age)
from app import app
app.run(port=args.port, debug=args.debug)
if __name__ == "__main__":
main()