seg_yinzy
/
myems
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

removed unused local variable in costfile.py of API

pull/61/MERGE
13621160019@163.com 2021-08-15 13:56:44 +08:00
parent 8372d3f741
commit 17dfe5d687
1 changed files with 32 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
myems-api/core/costfile.py
View File

@ -80,62 +80,60 @@ class CostFileCollection:
raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.USER_UUID_NOT_FOUND_IN_HEADERS_PLEASE_LOGIN') description='API.USER_UUID_NOT_FOUND_IN_HEADERS_PLEASE_LOGIN')
cnx = mysql.connector.connect(**config.myems_user_db) cnx_user_db = mysql.connector.connect(**config.myems_user_db)
cursor = cnx.cursor() cursor_user_db = cnx_user_db.cursor()
query = (" SELECT utc_expires " query = (" SELECT utc_expires "
" FROM tbl_sessions " " FROM tbl_sessions "
" WHERE user_uuid = %s AND token = %s") " WHERE user_uuid = %s AND token = %s")
cursor.execute(query, (user_uuid, token,)) cursor_user_db.execute(query, (user_uuid, token,))
row = cursor.fetchone() row = cursor_user_db.fetchone()
if row is None: if row is None:
if cursor: if cursor_user_db:
cursor.close() cursor_user_db.close()
if cnx: if cnx_user_db:
cnx.disconnect() cnx_user_db.disconnect()
raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_SESSION_PLEASE_RE_LOGIN') description='API.INVALID_SESSION_PLEASE_RE_LOGIN')
else: else:
utc_expires = row[0] utc_expires = row[0]
if datetime.utcnow() > utc_expires: if datetime.utcnow() > utc_expires:
if cursor: if cursor_user_db:
cursor.close() cursor_user_db.close()
if cnx: if cnx_user_db:
cnx.disconnect() cnx_user_db.disconnect()
raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.USER_SESSION_TIMEOUT') description='API.USER_SESSION_TIMEOUT')
cursor.execute(" SELECT id " cursor_user_db.execute(" SELECT id "
" FROM tbl_users " " FROM tbl_users "
" WHERE uuid = %s ", " WHERE uuid = %s ",
(user_uuid,)) (user_uuid,))
row = cursor.fetchone() row = cursor_user_db.fetchone()
if row is None: if row is None:
if cursor: if cursor_user_db:
cursor.close() cursor_user_db.close()
if cnx: if cnx_user_db:
cnx.disconnect() cnx_user_db.disconnect()
raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST', raise falcon.HTTPError(falcon.HTTP_400, title='API.BAD_REQUEST',
description='API.INVALID_USER_PLEASE_RE_LOGIN') description='API.INVALID_USER_PLEASE_RE_LOGIN')
else:
user_id = row[0]
cnx = mysql.connector.connect(**config.myems_historical_db) cnx_historical_db = mysql.connector.connect(**config.myems_historical_db)
cursor = cnx.cursor() cursor_historical_db = cnx_historical_db.cursor()
add_values = (" INSERT INTO tbl_cost_files " add_values = (" INSERT INTO tbl_cost_files "
" (file_name, uuid, upload_datetime_utc, status, file_object ) " " (file_name, uuid, upload_datetime_utc, status, file_object ) "
" VALUES (%s, %s, %s, %s, %s) ") " VALUES (%s, %s, %s, %s, %s) ")
cursor.execute(add_values, (filename, cursor_historical_db.execute(add_values, (filename,
file_uuid, file_uuid,
datetime.utcnow(), datetime.utcnow(),
'new', 'new',
raw_blob)) raw_blob))
new_id = cursor.lastrowid new_id = cursor_historical_db.lastrowid
cnx.commit() cnx_historical_db.commit()
cursor.close() cursor_historical_db.close()
cnx.disconnect() cnx_historical_db.disconnect()
resp.status = falcon.HTTP_201 resp.status = falcon.HTTP_201
resp.location = '/costfiles/' + str(new_id) resp.location = '/costfiles/' + str(new_id)