forked from achaorg/PyOne_Mongo
-
Notifications
You must be signed in to change notification settings - Fork 1
/
run.py
executable file
·115 lines (104 loc) · 4.03 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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
#-*- coding=utf-8 -*-
import eventlet
eventlet.monkey_patch()
import os
import traceback
from flask_script import Manager, Shell
from app import create_app
from self_config import *
from config import *
from function import *
from redis import Redis,ConnectionPool
app = create_app()
manager = Manager(app)
@app.cli.command()
def test():
"""Run the unit tests."""
import unittest
tests = unittest.TestLoader().discover('tests')
unittest.TextTestRunner(verbosity=2).run(tests)
######################系统日志
app.logger.addHandler(ErrorLogger().file_handler)
app.logger.setLevel(logging.DEBUG)
######################初始化变量
if REDIS_PASSWORD!="":
pool = ConnectionPool(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB,password=REDIS_PASSWORD)
else:
pool = ConnectionPool(host=REDIS_HOST, port=REDIS_PORT, db=REDIS_DB)
tmp_rd=Redis(connection_pool=pool)
try:
tmp_rd.set("allow_site",','.join(allow_site))
tmp_rd.set("downloadUrl_timeout",downloadUrl_timeout)
tmp_rd.set("password",password)
tmp_rd.set("title",title)
tmp_rd.set("tj_code",tj_code)
tmp_rd.set("headCode",headCode)
tmp_rd.set("footCode",footCode)
tmp_rd.set("cssCode",cssCode)
tmp_rd.set("robots",robots)
tmp_rd.set("theme",theme)
tmp_rd.set("title_pre",title_pre)
tmp_rd.set("redirect_uri",redirect_uri)
tmp_rd.set("BaseAuthUrl",BaseAuthUrl)
tmp_rd.set("app_url",app_url)
tmp_rd.set("ARIA2_HOST",ARIA2_HOST)
tmp_rd.set("ARIA2_PORT",ARIA2_PORT)
tmp_rd.set("ARIA2_SECRET",ARIA2_SECRET)
tmp_rd.set("ARIA2_SCHEME",ARIA2_SCHEME)
tmp_rd.set("MONGO_HOST",MONGO_HOST)
tmp_rd.set("MONGO_PORT",MONGO_PORT)
tmp_rd.set("MONGO_USER",MONGO_USER)
tmp_rd.set("MONGO_PASSWORD",MONGO_PASSWORD)
tmp_rd.set("MONGO_DB",MONGO_DB)
tmp_rd.set("REDIS_HOST",REDIS_HOST)
tmp_rd.set("REDIS_PORT",REDIS_PORT)
tmp_rd.set("REDIS_PASSWORD",REDIS_PASSWORD)
tmp_rd.set("REDIS_DB",REDIS_DB)
tmp_rd.set("show_secret",show_secret)
tmp_rd.set("default_sort",default_sort)
tmp_rd.set("order_m",order_m)
tmp_rd.set("encrypt_file",encrypt_file)
tmp_rd.set("default_pan",default_pan)
tmp_rd.set("admin_prefix",admin_prefix)
tmp_rd.set("balance",balance)
tmp_rd.set("thread_num",thread_num)
tmp_rd.set("verify_url",verify_url)
tmp_rd.set("show_doc",show_doc)
tmp_rd.set("show_image",show_image)
tmp_rd.set("show_video",show_video)
tmp_rd.set("show_dash",show_dash)
tmp_rd.set("show_audio",show_audio)
tmp_rd.set("show_code",show_code)
tmp_rd.set("show_redirect",show_redirect)
config_path=os.path.join(config_dir,'self_config.py')
with open(config_path,'r') as f:
text=f.read()
tmp_rd.set('users',re.findall('od_users=([\w\W]*})',text)[0])
key='themelist'
tmp_rd.delete(key)
except:
print('\033[31m redis鉴权失败！请注意修改！\033[0m')
######################函数
app.jinja_env.globals['version']=config.version
app.jinja_env.globals['FetchData']=FetchData
app.jinja_env.globals['path_list']=path_list
app.jinja_env.globals['CanEdit']=CanEdit
app.jinja_env.globals['quote']=urllib.quote
app.jinja_env.globals['len']=len
app.jinja_env.globals['enumerate']=enumerate
app.jinja_env.globals['breadCrumb']=breadCrumb
app.jinja_env.globals['list']=list
app.jinja_env.globals['os']=os
app.jinja_env.globals['re']=re
app.jinja_env.globals['file_ico']=file_ico
app.jinja_env.globals['CutText']=CutText
app.jinja_env.globals['GetConfig']=GetConfig
app.jinja_env.globals['config_dir']=config_dir
app.jinja_env.globals['GetThemeList']=GetThemeList
app.jinja_env.globals['GenerateToken']=GenerateToken
app.jinja_env.globals['get_od_user']=get_od_user
################################################################################
#####################################启动#######################################
################################################################################
if __name__ == '__main__':
manager.run()