if ('fund_name' in content and content['fund_name']) and ('bond_name' in content and content['bond_name']):
df = df[(df['fund_name'] == content['fund_name']) & (df['bond_name'] == content['bond_name'])]
elif 'fund_name' in content and content['fund_name']:
df = df[df['fund_name'] == content['fund_name']]
elif 'bond_name' in content and content['bond_name']:
df = df[df['bond_name'] == content['bond_name']]
else:
pass
from irFundpy import *
import datetime
def db_dml_batch(sql_query: str, val_list, datasource='tyircp'):
'''
:desc:批量插入
:param sql_query:
sql_query = "insert into definition_df_bck1 (SHEET_NO,C_TYPE) values (:1,:2)"
sql_query = "update definition_df_bck1 set C_TYPE= :1 where SHEET_NO= :2 "
sql_query='delete from definition_df_bck1 where sheet_no= :1'
:param val_list:->list,tuple
<参数顺序需与字段顺序一直>
[[]]
[(1,)] 或 [[1]]
[[1,2],[3,4]]或[(1,2),(3,4)]
:param datasource:
:return:
'''
try:
conn = get_connect(datasource)
conn_cursor = conn.cursor()
conn_cursor.executemany(sql_query, val_list)
conn.commit()
print('提交成功')
except Exception as e:
# 如果发生错误就进行回滚
conn.rollback()
print(e)
finally:
conn_cursor.close()
conn.close()
print('释放完成')
def db_dml(sql_query: str, datasource='tyircp'):
'''
:desc:单条数据插入
'''
try:
conn = get_connect(datasource)
conn_cursor = conn.cursor()
conn_cursor.execute(sql_query)
conn.commit()
print('提交成功')
except Exception as e:
# 如果发生错误就进行回滚
conn.rollback()
print(e)
finally:
conn_cursor.close()
conn.close()
def json_by_df(data: pd.DataFrame):
'''
数据转换,针对datetime的处理
数据转换,将df--->json数据[{"a": 0, "b": 1, "c": 2, "d": 3, "e": 4}, {"a": 5, "b": 6, "c": 7, "d": 8, "e": 9}]
'''
columns_keys = data.columns.values
columns_keys = list(columns_keys)
data_list = []
# 将date类型转化为str类型
for index, rows in data.iterrows():
tmp = list(rows)
for i in range(len((tmp))):
if isinstance(tmp[i], (datetime.datetime, datetime.date)):
data[columns_keys[i]] = data[columns_keys[i]].astype('str')
break
for index, rows in data.iterrows():
tmp = list(rows)
a = [{columns_keys[i]: tmp[i] for i in range(len(tmp))}]
data_list.append(a[0])
json_data = data_list
json_data = json.dumps(json_data, ensure_ascii=False)
# print(json_data)
return json_data
网友评论