refactor: first use sqlalchemy
parent
88dbaa0249
commit
e47da7bbff
@ -0,0 +1,22 @@
|
|||||||
|
'''
|
||||||
|
Desc:
|
||||||
|
File: /config.py
|
||||||
|
Project: config
|
||||||
|
File Created: Sunday, 14th August 2022 5:57:02 pm
|
||||||
|
Author: luxuemin2108@gmail.com
|
||||||
|
-----
|
||||||
|
Copyright (c) 2022 Camel Lu
|
||||||
|
'''
|
||||||
|
import os
|
||||||
|
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
load_dotenv()
|
||||||
|
|
||||||
|
env_db_host = os.getenv('db_host')
|
||||||
|
env_db_port = os.getenv('db_port')
|
||||||
|
env_db_name = os.getenv('db_name')
|
||||||
|
|
||||||
|
env_db_user = os.getenv('db_user')
|
||||||
|
env_db_password = os.getenv('db_password')
|
||||||
|
|
||||||
|
env_db_stock_name = os.getenv('db_stock_name')
|
@ -0,0 +1,36 @@
|
|||||||
|
'''
|
||||||
|
Desc:
|
||||||
|
File: /engine.py
|
||||||
|
Project: db
|
||||||
|
File Created: Sunday, 14th August 2022 5:41:01 pm
|
||||||
|
Author: luxuemin2108@gmail.com
|
||||||
|
-----
|
||||||
|
Copyright (c) 2022 Camel Lu
|
||||||
|
'''
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
sys.path.append(os.getcwd() + '/src')
|
||||||
|
|
||||||
|
from sqlalchemy import create_engine, text
|
||||||
|
from config.env import env_db_host, env_db_name, env_db_user, env_db_password, env_db_port
|
||||||
|
|
||||||
|
SQLALCHEMY_DATABASE_URI = 'mysql+pymysql://' + env_db_user + ':'+ env_db_password +'@'+ env_db_host + ':' + env_db_port + '/' + env_db_name
|
||||||
|
|
||||||
|
_global_engine = None
|
||||||
|
|
||||||
|
def get_engine():
|
||||||
|
global _global_engine
|
||||||
|
if _global_engine == None:
|
||||||
|
_global_engine = create_engine(SQLALCHEMY_DATABASE_URI, echo=True, future=True)
|
||||||
|
return _global_engine
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
engine1 = get_engine()
|
||||||
|
with engine1.connect() as conn:
|
||||||
|
query = {"fund_code": '000696'}
|
||||||
|
result = conn.execute(
|
||||||
|
text("SELECT * FROM fund_morning_snapshot WHERE fund_code = :fund_code"),
|
||||||
|
query
|
||||||
|
)
|
||||||
|
for row in result:
|
||||||
|
print("row", row)
|
Loading…
Reference in New Issue