def main():
# server = ThinkSocks()
# server.listen(8530)
# g_logger.info("Server Started!!!")
# tornado.ioloop.IOLoop.instance().start()
# multiple sub-process
# server = ThinkSocks()
# server.bind(8530)
# server.start(0) # Forks multiple sub-processes
# g_logger.info("Server Started!!!")
# tornado.ioloop.IOLoop.current().start()
# adv multiple sub-process
sockets = bind_sockets(8530)
tornado.process.fork_processes(0)
from ThinkSocks.ThinkSocks import ThinkSocks
server = ThinkSocks()
server.add_sockets(sockets)
from pythinkutils.common.log import g_logger
g_logger.info("Server Started!!!")
IOLoop.current().start()
def main():
g_logger.info(g_config.get("mysql", "host"))
parser = argparse.ArgumentParser()
parser.add_argument("--env", type=str, default="pro", help="--env prd/dev")
args = parser.parse_args()
print(args)
async def test_get_group():
dictRet = await GroupService.get_group_by_name("admin")
if dictRet is None:
g_logger.info("FXXK")
return
g_logger.info(obj2json(dictRet))
def mk_chrome(self, bHeadless=True):
option = webdriver.ChromeOptions()
if bHeadless:
option.add_argument('headless')
option.add_argument('--no-sandbox')
szUA = self.on_get_ua()
if False == is_empty_string(szUA):
option.add_argument("user-agent={0}".format(szUA))
szProxy = self.on_get_proxy()
if False == is_empty_string(szProxy):
g_logger.info("--proxy-server={}".format(szProxy))
option.add_argument("--proxy-server={}".format(szProxy))
else:
pass
option.add_argument('--user-data-dir={}'.format(
self.on_get_data_dir()))
driver = webdriver.Chrome(options=option)
driver.delete_all_cookies()
driver.set_window_size(1080, 1920)
driver.set_page_load_timeout(60)
self.m_eventDriver = EventFiringWebDriver(driver, self)
async def test_create_permission():
await PermissionService.create_permission("permission_read_user_info")
dictPermission = await PermissionService.select_permission("permission_read_user_info")
g_logger.info(dictPermission)
await PermissionService.grant_permission_to_user(dictPermission["id"], 10000001)
await PermissionService.grant_permission_to_group(dictPermission["id"], 10000001)
async def test_create_user():
szUsername = "******"
szPwd = "Ab123456"
dictRet = await SimpleUserService.create_user(szUsername, szPwd)
g_logger.info(dictRet)
ret = await SimpleUserService.login(szUsername, szPwd)
g_logger.info(ret)
def get_deal_by_month():
nStart = get_timestamp()
nCount, lstData = DealService.get_deal_by_month("2017-07",
nOffset=0,
nLimit=10)
g_logger.info(
"%d => %s" %
(nCount, obj2json(lstData).encode('utf-8').decode('unicode_escape')))
g_logger.info("%d s" % (get_timestamp() - nStart, ))
def main():
# nStartTime = get_timestamp()
# g_logger.info("[%d s] %d / %d" % ((get_timestamp() - nStartTime), get_offset(), g_nTotal))
g_logger.info("Start!!!")
while get_offset() < g_nTotal:
try:
if insert_deal():
continue
else:
time.sleep(10)
except Exception as e:
g_logger.info(e)
time.sleep(10)
continue
def insert_deal():
nStartTime = get_timestamp()
conn = ThinkPG.get_conn_pool_ex().getconn()
cur = conn.cursor()
try:
cur.execute(
'''
insert into public.t_deal
select * from csv.t_deal offset %s limit %s
''', (get_offset(), g_nLimit))
conn.commit()
incr_offset()
g_logger.info("[%d s] %d / %d" %
((get_timestamp() - nStartTime), get_offset(), g_nTotal))
return True
except Exception as e:
g_logger.info(e)
return False
finally:
ThinkPG.get_conn_pool_ex().putconn(conn)
async def test_create_group():
dictRet = await GroupService.create_group("FXXK3")
if dictRet is None:
g_logger.info("FXXK Failed")
return
g_logger.info(obj2json(dictRet))
ret = await GroupService.add_user_to_group(10000008, dictRet["id"])
g_logger.info(ret)
async def test_login():
szUsername = "******"
szPwd = "Ab123456"
ret = await SimpleUserService.login(szUsername, szPwd)
g_logger.info(ret)
def after_navigate_to(self, url, browser):
g_logger.info("After navigate to %s" % (url, ))
def before_navigate_to(self, url, browser):
g_logger.info("Before navigate to %s" % (url, ))
async def on_goto_login_page(self):
g_logger.info("Goto login page")
self.redirect("/login")
async def test_user_token():
ret = await SimpleUserService.check_token(
10000008, "85b06a75bf6b4e0aa0d759dabb457130")
g_logger.info(ret)
from pythinkutils.aio.auth.tornado.handler.ThinkLoginHandler import ThinkLoginHandler
from handler.LogoutHandler import LogoutHandler
from handler.Testhandler import TestHandler
application = tornado.web.Application(
handlers=[(r"/login", ThinkLoginHandler), (r"/logout", LogoutHandler),
(r"/test", TestHandler), (r'/', MainHandler)],
compress_response=True,
cookie_secret="BUEa2ckrQtmBofim3aP6cwr/acg0LEu6mHUxq4O3EY0=",
template_path=os.path.join(os.path.dirname(__file__), "templates")
# 静态文件
,
static_path=os.path.join(os.path.dirname(__file__), "static"),
autoreload=False)
async def on_server_started():
g_logger.info("Server Started!")
if __name__ == '__main__':
http_server = HTTPServer(application)
http_server.bind(8590)
http_server.start(0)
# ipDB = IPLocation.instance()
g_logger.info('HTTP Server started... %d' % (os.getpid(), ))
asyncio.gather(on_server_started())
tornado.ioloop.IOLoop.current().start()
async def group_has_permission():
dictPermission = await PermissionService.select_permission("permission_read_user_info")
bHasPermission = await PermissionService.group_has_permission(10000001, dictPermission["id"])
g_logger.info(bHasPermission)
async def on_goto_login_page(self):
g_logger.info("Goto login page")
self.redirect("/login?redirect_url=%2ffxxk")
async def on_server_started():
szMsg = await func1()
g_logger.info(szMsg)
async def on_server_started():
g_logger.info("Server Started!")
def main():
while True:
g_logger.info("%s" % (get_current_time_str()))
time.sleep(5)
