本文介绍了一个使用Python的MySQL连接库pymysql实现的函数，通过`common_functions`类执行SQL脚本获取数据。核心内容是`calculate_sharecycle`方法，用于计算特定日期范围内股票的交易周期。在`gen_report.py`中，通过调用这些函数获取并打印了分享循环次数。
------------------ comm_functions.py ------------
import pymysql
class common_functions(object):
# ---- 需要初始化 __init__ 来连接数据库，并产生 self.cur 供后面函数应用
def __init__(self):
self.conn = pymysql.connect(
host='192.168.68.189',
port=3307,
user='xxxxx',
passwd='xxxx',
db='xxxx',
)
self.cur = self.conn.cursor()
def execute_sqlscript_fetchall_contents(self, sqlscript):
self.cur.execute(sqlscript)
contents = self.cur.fetchall()
return contents
def execute_sqlscript_fetchall_content(self, sqlscript):
contents = self.execute_sqlscript_fetchall_contents(sqlscript)
for row in contents:
data = row[0]
return data
def calculate_sharecycle(self, open_time, clean_time):
sql_sharecycle = "select count(*) from hd_trading_date where open_day >= left(" + open_time + ", 8) and open_day <= left(" + clean_time + ",8)"
sharecycle = self.execute_sqlscript_fetchall_content(sql_sharecycle)
if open_time[8:12] >= "2100":
sharecycle = sharecycle - 1
if clean_time[3][8:12] >= "2100":
sharecycle = sharecycle + 1
------- gen_report.py ----------
import sys
sys.path.append("./sqldata")
from com_functions import common_functions
def get_data():
sharecycle = common_functions().calculate_sharecycle("20171018143914", "20171018210220")
print("sharecycle: %d" % sharecycle)
if __name__ == "__main__":
get_data()
common_functions().close_sql()
扫一扫
1486
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
