def post(self, request, format=None):
s = MysqlClusterSerializer(data=request.data)
foreign_ip = request.data['foreign_ip']
foreign_port = request.data['foreign_port']
plat_user = request.data['plat_user']
plat_user_passwd = request.data['plat_user_pass']
msg = ""
if s.is_valid(): # 验证
try:
testconn = MySQL(ip=foreign_ip,
port=foreign_port,
user=plat_user,
passwd=plat_user_passwd)
testconn.close()
msg = '数据库连接成功'
except Exception as e:
print(e)
json_data = {'code': 500, 'msg': '数据添加失败，测试连接失败'}
return Response(json_data, content_type="application/json")
s.save() #连接成功后保存数据
json_data = {'code': 200, 'msg': '数据保存成功,' + msg}
json_data['data'] = s.data
return Response(json_data, content_type="application/json")
# return api_response.JsonResponse(s.data,code=status.HTTP_200_OK,msg='success')
json_data = {'code': 500, 'msg': '数据添加失败，请检查数据格式'}
return Response(json_data, content_type="application/json")
def put(self, request, format=None):
try:
data_id = request.data['id']
foreign_ip = request.data['foreign_ip']
foreign_port = request.data['foreign_port']
plat_user = request.data['plat_user']
plat_user_passwd = request.data['plat_user_pass']
except Exception as e:
print(e)
json_data = {'code': 500, 'msg': '数据有错误获取不到id'}
return Response(json_data,status=500)
else:
DATA_MODEL = self.get_object(data_id)
s = MysqlClusterSerializer(DATA_MODEL, data=request.data)
if s.is_valid(raise_exception=True):
try:
testconn = MySQL(ip=foreign_ip, port=foreign_port, user=plat_user, passwd=plat_user_passwd)
testconn.close()
msg = '数据库连接成功'
s.save()
json_data = {'code': 200, 'msg': '更新成功'}
return Response(json_data)
except Exception as e:
json_data = {'code': 500, 'msg': '更新失败，新的用户名密码或者网络不通过'}
return Response(json_data)
json_data = {'code': 500, 'msg': '更新失败,数据格式和内部表不一致'}
return Response(json_data, status=status.HTTP_500_INTERNAL_SERVER_ERROR)
class MySQLTask(TaskThread):
def __init__(self, *args, **kwargs):
self._mysql = MySQL(**kwargs["db_config"])
super(MySQLTask,self).__init__(*args, **kwargs)
self.daemon = True
def get_handler(self):
return self._mysql
def close(self):
self._mysql.close()
for item in arr:
if item == "safe" or item == "unsure":
temp = scoreDict[item]
ret.append((pd, temp))
else:
ret.append((pd, None))
for item in ret:
if item[1] == None:
sql_update = "UPDATE domain_static SET credit=NULL WHERE primary_domain='%s';" % item[
0]
else:
sql_update = "UPDATE domain_static SET credit=%d WHERE primary_domain='%s';" % (
item[1], item[0])
dns_db.execute(sql_update)
if __name__ == "__main__":
try:
read_data()
# create_db()
# init_db()
# add_register_years()
add_credit_evidence()
except Exception as e:
print(e, time.asctime(time.localtime(time.time())))
finally:
dns_db.close()
