def __init__(self, info):
self.hostip = info[0]
self.port = int(info[1])
self.username = info[2]
self.password = info[3]
# servername = info[4]
if info[5]:
self.remote_dir = info[5]
#xwdm检查专用列
if info[6]:
self.xwdm_check_col = info[6]
if info[7]:
self.follow_hostip = info[7]
if info[8]:
self.follow_port = int(info[8])
if info[9]:
self.follow_username = info[9]
if info[10]:
self.follow_password = info[10]
if info[11]:
self.follow_dir = info[11]
#获得当天日期字符串
self.local_date = dt.datetime.today().strftime('%Y%m%d')
self.next_date = (dt.datetime.today()+dt.timedelta(1)).strftime('%Y%m%d')
self.sshClient = ct.sshConnect(self.hostip, self.port, self.username, self.password)
def check_follow(self):
#先清除历史文件/home/trade/run/timaker_hx/follow
t = paramiko.Transport((self.hostip, self.port))
t.connect(username=self.username, password=self.password)
sftp = paramiko.SFTPClient.from_transport(t)
self.rm_remote_dir(sftp, self.remote_dir)
#上传跟投文件到服务器
#scp [email protected]:/home/trade/csvfiles/FollowSecurity_YYYYMMDD.csv /home/trade/run/timaker_hx/follow
#command = "scp " + self.follow_username + "@" + self.follow_hostip + ":" + self.follow_dir + self.next_date + "* " + self.remote_dir
#command = "scp " + self.follow_username + "@" + self.follow_hostip + ":" + self.follow_dir + "FollowSecurity_" + "\*$" + self.next_date + ".csv " + self.remote_dir
command = "scp " + self.follow_username + "@" + self.follow_hostip + ":" + self.follow_dir + "FollowSecurity_" + self.next_date + ".csv " + self.remote_dir
logger.info(command)
sshRes = ct.sshExecCmd(self.sshClient, command)
# print("sshRes:",sshRes)
ct.sshClose(self.sshClient)
#等待2s防止文件没有上传成功
time.sleep(2)
#匹配文件，并校验文件内容是否有重复的记录
sshClient = ct.sshConnect(self.hostip, self.port, self.username, self.password)
#command2 = "ls " + self.remote_dir + self.next_date + "*"
command2 = "ls '" + self.remote_dir + "FollowSecurity_" + self.next_date + ".csv'"
logger.info(command2)
sshRes2 = ct.sshExecCmd(sshClient, command2)
#print("sshRes2:",sshRes2)
if len(sshRes2)==1:
ffilename = sshRes2[0]
command3 = "cat " + ffilename + " | awk -F\",\" \'{OFS=\",\";print $1}\'"
#command3 = "cat " + "'" + self.remote_dir + "FollowSecurity_$" + self.next_date + ".csv'" + " | awk -F\",\" \'{OFS=\",\";print $1}\'"
sshRes3 = ct.sshExecCmd(sshClient, command3)
#print("sshRes3:",sshRes3)
if sshRes3 == []:
msg = "Error: 跟投文件[%s]内容为空" % ffilename
logger.warning(msg)
filename = msg
else:
df = pd.DataFrame(sshRes3[1:],columns={sshRes3[0]})
#print("length:", len(df))
if len(df) != 0 :
if len(df[df.duplicated()]) == 0:
filename = sshRes2[0]
else:
dup_df = df[df.duplicated()]
dup_list = list(dup_df[sshRes3[0]])
dup_str = ','.join(dup_list)
msg = "跟投文件[%s]有重复的证券代码记录[%s]" % (ffilename, dup_str)
logger.warning(msg)
filename = "Error: " + msg
else:
msg = "Error: 跟投文件[%s]内容为空" % ffilename
logger.warning(msg)
filename = msg
elif len(sshRes2)==0:
#temstr = ",".join(sshRes2)
filename = "Error:没有匹配到跟投文件"
else:
filename = "Error:跟投文件匹配多个，" + ",".join(sshRes2)
ct.sshClose(sshClient)
return filename
def __init__(self, info):
self.hostip = info[0]
self.port = int(info[1])
self.username = info[2]
self.password = info[3]
# servername = info[4]
self.remote_dir = info[5]
#获得当天日期字符串
self.local_date = dt.datetime.today().strftime('%Y-%m-%d')
self.sshClient = ct.sshConnect(self.hostip, self.port, self.username,
self.password)
def ssh_remote_command_task():
error_list = []
linuxInfo = ct.get_server_config('./config/ssh_remote_command_config.txt')
for info in linuxInfo:
hostip = info[0]
port = int(info[1])
username = info[2]
password = info[3]
command = info[5]
sshClient = ct.sshConnect(hostip, port, username, password)
if sshClient == 999:
msg = "Failed:服务器[%s],连接失败，请检查密码是否正确" % hostip
logger.error(msg)
#ct.send_sms_control("NoLimit", msg)
error_list.append(msg)
else:
logger.info("Ok: 服务器[%s]连接正常" % hostip)
logger.info(hostip + "::" + command)
#sshRes = ct.sshExecCmd(sshClient, command)
stdin, stdout, stderr = sshClient.exec_command(command)
stdoutstr = stdout.read().decode('utf-8')
ssherr = stderr.read().decode('utf-8')
if ssherr:
msg = "服务器[%s]ssh执行命令返回错误：[%s]" % (hostip, ssherr)
logger.warning(msg)
error_list.append(msg)
sshRes = []
sshRes = stdoutstr.strip().split('\n')
if sshRes == ['']:
sshRes = []
logger.info("sshRes:")
logger.info(sshRes)
sshClient.close()
if len(error_list) != 0:
temstr = ';'.join(error_list)
msg = "Failed:服务器ssh执行命令失败列表：[%s]" % temstr
logger.error(msg)
ct.send_sms_control("NoLimit", msg)
else:
logger.info("所有服务器ssh执行命令正常！")
def get_query_data(linuxInfo):
logger = logging.getLogger()
yaml_path = './config/non_trade_monitor_logger.yaml'
ct.setup_logging(yaml_path)
for info in linuxInfo:
hostip = info[0]
port = info[1]
username = info[2]
password = info[3]
# servername = info[4]
command = info[5]
cur_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
temstr = "**********" + cur_time + "::" + hostip + "::" + command + "::**********"
ct.write_file(query_result_file, temstr)
sshClient = ct.sshConnect(hostip, port, username, password)
sshRes = ct.sshExecCmd(sshClient, command)
logger.info(hostip + "::" + command)
try:
for item in sshRes:
# de_item = item.decode('gb2312')
# error_list = de_item.strip().split(':', 1)
# grep_lists.append(error_list)
# memstr=','.join(error_list)
# print memstr
# temstr= item.strip().encode('utf-8')
temstr = item.strip()
logger.info(temstr)
ct.write_file(query_result_file, temstr)
except Exception as e:
msg = "write failed: [hostip:%s];[username:%s];[error:%s]" % (
hostip, username, str(e))
logger.error(msg)
ct.write_log(log_file, msg)
ct.sshClose(sshClient)
logger.info("get_query_data finished")
for handler in logger.handlers:
logger.removeHandler(handler)
def newserver_employ_task():
error_list = []
linuxInfo = ct.get_server_config('./config/newserver_employ_config.txt')
for info in linuxInfo:
hostip = info[0]
port = int(info[1])
username = info[2]
password = info[3]
#command = info[5]
command = "/tmp/employ_server_package/employment_script.sh"
#上传部署服务器./employ_server_package到新服务器的tmp目录
ft.upload("./employ_server_package", "/tmp/employ_server_package", hostip, port, username, password, '')
#执行shell脚本部署环境
sshClient = ct.sshConnect(hostip, port, username, password)
if sshClient == 999:
msg = "Failed:服务器[%s],连接失败，请检查密码是否正确" % hostip
logger.error(msg)
#ct.send_sms_control("NoLimit", msg)
error_list.append(msg)
else:
logger.info("Ok: 服务器[%s]连接正常" % hostip)
#执行shell脚本部署环境
logger.info(hostip + "::" + command)
# sshRes = ct.sshExecCmd(sshClient, command)
stdin, stdout, stderr = sshClient.exec_command(command)
stdoutstr = stdout.read().decode('utf-8')
ssherr = stderr.read().decode('utf-8')
if ssherr:
msg = "服务器[%s]ssh执行命令返回错误：[%s]" % (hostip, ssherr)
logger.debug(msg)
#error_list.append(msg)
sshRes = []
sshRes = stdoutstr.strip().split('\n')
if sshRes == ['']:
sshRes = []
logger.info("sshRes:")
logger.info(sshRes)
sshClient.close()
def check_ssh_connect_task():
error_list = []
linuxInfo = ct.get_server_config('./config/check_root_passwd_config.txt')
for info in linuxInfo:
hostip = info[0]
port = int(info[1])
username = info[2]
password = info[3]
os_flag = info[4]
if os_flag == 'l':
sshClient = ct.sshConnect(hostip, port, username, password)
if sshClient == 999:
msg = "Failed:服务器[%s],连接失败，请检查密码是否正确" % hostip
logger.error(msg)
#ct.send_sms_control("NoLimit", msg)
error_list.append(hostip + ":::" + username + ":::" + password)
else:
logger.info("Ok: 服务器[%s]连接正常" % hostip)
sshClient.close()
else:
os_info = ct.get_remote_windows_os_info(username, password, hostip)
if os_info == 'Null':
msg = "Failed:服务器[%s],连接失败，请检查密码是否正确" % hostip
logger.error(msg)
#ct.send_sms_control("NoLimit", msg)
error_list.append(hostip + ":::" + username + ":::" + password)
else:
logger.info("Ok: 服务器[%s]连接正常,操作系统是[%s]" % (hostip,os_info))
if len(error_list) != 0:
temstr = ';'.join(error_list)
msg = "Failed:服务器连接失败列表：[%s]，请检查密码是否正确" % temstr
logger.error(msg)
ct.send_sms_control("NoLimit", msg)
else:
logger.info("所有服务器连接正常！")
def check_exchange_file(self):
#20200515,取消复制和检查SJSGB文件
# #1，复制windows文件到linux服务器1
# last_workday_i = ct.getLastWorkDay()
# last_workday = last_workday_i.replace('-','')
# file_date_str = last_workday[-4:]
# logger.info("last_workday:" + last_workday)
# winserver = self.follow_hostip
# admin_passwd = self.follow_password
# admin_passwd = "adminadmin\$8"
# sjs_file_dir = "/home/trade/ExchFile/sjs_file/"
# copy_file_name = "SJSGB" + file_date_str + ".DBF"
# new_file_name = "SJSGB" + self.local_date[-4:] + ".DBF"
# #print(new_file_name)
# win_file_local_path = "/D:/tora/back_cmd/sjs_file/" + copy_file_name
# sjs_back_remote_path = sjs_file_dir + new_file_name
# command = '%sscp_task.sh %s %s %s %s %s' % (sjs_file_dir,winserver,'administrator',admin_passwd,win_file_local_path,
# sjs_back_remote_path)
# logger.info("cp_file_Command:" + command)
# #本地执行
# #com_res = os.system(command)
# try:
# ret = subprocess.run(command,shell=True,stdin=subprocess.PIPE,
# stdout=subprocess.PIPE,stderr=subprocess.PIPE,universal_newlines=True,timeout=10,check=False)
# com_res = ret.returncode
# logger.info("com_res:" + str(com_res))
# logger.info("ret.stdout:")
# logger.info(ret.stdout)
# logger.info("ret.stderr:")
# logger.info(ret.stderr)
# except Exception as e:
# msg = "复制文件异常：" + str(e)
# com_res = 256
# logger.error(msg)
# #检查文件是否在
# command_check = "ls " + sjs_back_remote_path
# logger.info("command_check:" + command_check)
# check_res = subprocess.run(command_check,shell=True,stdin=subprocess.PIPE,
# stdout=subprocess.PIPE,stderr=subprocess.PIPE,universal_newlines=True,timeout=10,check=False)
# logger.info("check_copy_file:")
# logger.info(check_res.stdout)
# #如果执行失败的话，再执行一次。
# if com_res != 0 or check_res.stdout=='':
# logger.info("Failed:从服务器[%s]复制文件[%s]第一次失败" % (winserver,win_file_local_path))
# ret2 = subprocess.run(command,shell=True,stdin=subprocess.PIPE,
# stdout=subprocess.PIPE,stderr=subprocess.PIPE,universal_newlines=True,timeout=10,check=False)
# com_res2 = ret2.stdout
# logger.info("第二次执行结果com_res2: " + str(com_res2))
# else:
# logger.info("Ok:从服务器[%s]复制文件[%s]成功" % (winserver,win_file_local_path))
# #从linux1复制scp到linux2,linux3，要先做scp免密认证，从目的IP到本地7的免密验证。
# #scp [email protected]:/home/trade/csvfiles/FollowSecurity_YYYYMMDD.csv /home/trade/run/timaker_hx/follow
# for linux_r_ip in ["10.188.80.16","192.168.253.197","10.188.80.67","192.168.253.135"]:
# #linux_r_ip = '192.168.238.7'
# sshClient_r = ct.sshConnect(linux_r_ip, self.port, self.username, self.password)
# #linux_remote = '/home/trade/ExchFile/sjs_fie/'
# command_linux_r = "scp " + self.username + "@" + self.hostip + ":" + sjs_back_remote_path + " " + sjs_back_remote_path
# logger.info(command_linux_r)
# sshRes = ct.sshExecCmd(sshClient_r, command_linux_r)
# # print("sshRes:",sshRes)
# #ct.sshClose(sshClient_r)
# #等待2s防止文件没有上传成功
# time.sleep(2)
# #匹配文件，并校验文件内容是否有重复的记录
# #command2 = "ls " + self.remote_dir + self.next_date + "*"
# command2 = "ls " + sjs_back_remote_path
# logger.info(command2)
# sshRes2 = ct.sshExecCmd(sshClient_r, command2)
# ct.sshClose(sshClient_r)
# if len(sshRes2)==0 :
# msg = "服务器[%s],文件[%s]没有匹配到，复制失败！" % (linux_r_ip, new_file_name)
# logger.error(msg)
# ct.send_sms_control('NoLimit',msg)
# else:
# logger.info("服务器[%s],文件[%s]匹配到，复制成功！" % (linux_r_ip, new_file_name))
#2，检查交易所基础文件是否存在，比对文件大小是否为0
sse_file_dir = "/home/trade/ExchFile/sse/"
sse_file = ["cpxx0201{mmdd}.txt", "fjy{tradeday}.txt", "gzlx.{Mdd}", "kxx{mmdd}.txt", "xzsl{mmdd}.txt",
"dbp{mmdd}.txt", "sfpm01{mmdd}.txt"]
szse_file_dir = "/home/trade/ExchFile/szse/"
szse_file = ["securities_{tradeday}.xml", "cashauctionparams_{tradeday}.xml", "issueparams_{tradeday}.xml",
"rightsissueparams_{tradeday}.xml","securityswitch_{tradeday}.xml", "imcparams_{tradeday}.xml",
"imcsecurityparams_{tradeday}.xml", "imcexchangerate_{tradeday}.xml","hkexreff04_{tradeday}.txt",
"hkexzxjc_{tradeday}.txt"]
mmdd = self.local_date[-4:]
tradeday = self.local_date
Mdd = hex(int(self.local_date[-4:-2]))[-1] + self.local_date[-2:]
new_sse_file_name=[]
new_szse_file_name=[]
#20200515,取消复制和检查SJSGB文件
#check_file_list = [sjs_back_remote_path]
check_file_list = []
for file_name in sse_file:
cc = file_name.replace("{mmdd}",mmdd).replace("{Mdd}",Mdd).replace("{tradeday}",tradeday)
full_path = sse_file_dir + cc
new_sse_file_name.append(cc)
check_file_list.append(full_path)
print(new_sse_file_name)
for file_name in szse_file:
cc = file_name.replace("{tradeday}",tradeday)
full_path = szse_file_dir + cc
new_szse_file_name.append(cc)
check_file_list.append(full_path)
print(new_szse_file_name)
#file_list = ["error_log_","download"]
error_file_list = []
check_server_list = ["10.188.80.16","192.168.253.197"]
for check_server_ip in check_server_list:
self.sshClient = ct.sshConnect(check_server_ip, self.port, self.username, self.password)
for file_path in check_file_list:
#file_path = self.remote_dir + file_pre + self.local_date + ".txt"
try:
file_size = ct.get_remote_filesize(self.sshClient,file_path)
if file_size != '0':
logger.info("Ok: 服务器 %s 文件 %s 的大小为 %s ，检查交易所基础数据成功!" % (check_server_ip, file_path, file_size))
else:
msg = "Error: 盘前交易所基础文件检查失败，服务器 %s 文件 %s 的大小为0或者文件不存在！" % (check_server_ip, file_path)
logger.error(msg)
error_file_list.append(file_path)
except Exception as e:
msg = "服务器 %s 获取交易所基础文件%s 大小失败，错误信息： %s" % (check_server_ip, file_path, str(e))
logger.error(msg)
error_file_list.append(file_path)
ct.sshClose(self.sshClient)
return error_file_list
def check_remote_csv_ZJXX(info, cursor, conn):
# server = '192.168.238.10'
# user = '******'
# password = '******'
# database = 'download'
# conn = pymssql.connect(server, user, password, database)
# cursor = conn.cursor()
# sship = "192.168.238.7"
# sshport = 22
# sshuser = "******"
# sshpw = "trade"
dbname = info["dbname"]
serverip = info["serverip"]
sship = info['sship']
sshport = int(info['sshport'])
sshuser = info['sshuser']
sshpw = info['sshpw']
csv_remote_dir_i = info['csv_remote_dir']
csv_remote_dir = csv_remote_dir_i.replace('{ndates}', ndates)
check_flag = False
#获取csv文件中的ZJZH,KYJE
zjxxfile_path = csv_remote_dir + '/' + "VIP_ZJXX" + ndates + ".csv"
command = "cat " + zjxxfile_path + " | awk -F\",\" \'{OFS=\",\";print $3,$7}\'"
# zjxxfile_path = '/home/trade/temp/debugkhh.csv'
# command = "cat " + zjxxfile_path + " | awk -F\",\" \'{OFS=\",\";print $1,$2}\'"
logger.debug("command:" + command)
sshClient = ct.sshConnect(sship, sshport, sshuser, sshpw)
sshRes = []
sshRes = ct.sshExecCmd(sshClient, command)
ct.sshClose(sshClient)
logger.debug("sshRes:")
logger.debug(sshRes)
#去掉'\r'
tem_ssh = []
for item in sshRes:
item = item.replace('\r', '')
tem_ssh.append(item)
if len(tem_ssh) != 0:
zjxx_columns = tem_ssh[0].split(',')
zjxx_data = []
for ssr in tem_ssh[1:]:
lists = ssr.split(',')
# print("lists:", lists)
zjxx_data.append(lists)
zjxx_df = pd.DataFrame(zjxx_data, columns=zjxx_columns)
zjxx_df.set_index('ZJZH', inplace=True)
# for index, row in zjxx_df.iterrows():
# print index, row['KYJE']
logger.debug("The csv KYJE:")
logger.debug(zjxx_df)
#把结尾是'11'的资金账号过滤掉
index_filter = filter(lambda x: x[-2:] != '11', zjxx_df.index)
logger.debug('index_filter:')
logger.debug(index_filter)
zjxx_filter_df = zjxx_df.loc[index_filter]
logger.debug("zjxx_filter_df")
logger.debug(zjxx_filter_df)
#df.loc[df['column_name'] != some_value]
#获取数据库数据
sql = "SELECT [AccountID],[UsefulMoney],[FrozenCash],[FrozenCommission] FROM [" + dbname + "].[dbo].[t_TradingAccount]"
(res, des) = mt.only_fetchall(cursor, conn, sql)
# res_list=[]
db_columns = list(zip(*des))[0]
temlist = [str(i) for i in db_columns]
db_kyje_df = pd.DataFrame(res, columns=temlist)
db_kyje_df = db_kyje_df.rename(columns={'AccountID': 'ZJZH'})
db_kyje_df.set_index('ZJZH', inplace=True)
#UserfulMoney列加上FrozenCash和FrozenCommission冻结字段
db_kyje_df['UsefulMoney'] = db_kyje_df.apply(lambda x: x[
'UsefulMoney'] + x['FrozenCash'] + x['FrozenCommission'],
axis=1)
new_df = pd.merge(zjxx_filter_df, db_kyje_df, how='left', on='ZJZH')
logger.debug("new_df:")
logger.debug(new_df)
new_df['KYJE'] = new_df['KYJE'].map(lambda x: round(float(x), 2))
new_df['UsefulMoney'] = new_df['UsefulMoney'].map(
lambda x: round(covervalue(x), 2))
money_diff = new_df.diff(axis=1)
#过滤出差别不为0的客户
def_khje = money_diff.loc[money_diff['UsefulMoney'] != 0]
if len(def_khje) != 0:
check_count = 0
def_khje = def_khje.drop(columns='KYJE')
def_khje = def_khje.rename(columns={'UsefulMoney': 'InOutMoney'})
for index, row in def_khje.iterrows():
logger.info(
"UsefulMoney not equal KYJE, the ZJZH is: %s, Difference is: %.2f"
% (index, round(row['InOutMoney'], 2)))
TransferMoney = round(
get_cust_TransferMoney(info, index, cursor, conn), 2)
logger.info("ZJZH %s transfer money is: %.2f" %
(index, TransferMoney))
final_diff = TransferMoney - row['InOutMoney']
#因为每笔委托会冻结0.1元，不体现在总冻结里,额外冻结资金设置一个数，小于这个数不报警，认为这个是被委托冻结了。
extra_frozen = 10.0
#比较资金差额和出入金，一致的话则没有问题。
if final_diff == 0:
logger.info("The Difference equal to TransferMoney")
check_count += 1
#因为每笔委托会冻结0.1元，不体现在总冻结里，所以要单独处理。
elif final_diff > 0 and final_diff < extra_frozen:
logger.info("The Difference extra_frozen %.2f" %
final_diff)
check_count += 1
else:
logger.error(
"error:The csv ZJZH: %s KYJE not equal to DB: %s UsefulMoney,Difference is: %.2f"
% (index, serverip, round(row['InOutMoney'], 2)))
check_flag = (check_count == len(def_khje))
else:
logger.info("ok:The csv KYJE equal to DB UsefulMoney")
check_flag = True
else:
logger.error(
"error: Failed to get remote server: %s csv file: %s, please check the config!"
% (sship, zjxxfile_path))
return check_flag