diff --git a/handler/company.py b/handler/company.py index 0eeb2cd..32f7766 100644 --- a/handler/company.py +++ b/handler/company.py @@ -33,10 +33,14 @@ class Company(Resource): data = mydb.query(sql) log.info(f"get data from db was {data}") - company = {} - company['id'], company['name'], company['contact'], company['appid'], company['appkey'] = data - company_redis.hmset(company['id'], company) - company_redis.expire(company['id'], expire_time) + if data: + company = {} + company['id'], company['name'], company['contact'], company['appid'], company['appkey'] = data[0] + company_redis.hmset(company['id'], company) + company_redis.expire(company['id'], expire_time) + else: + log.error(f"{comp_id} not found in mysql !") + return jsonify({'code': 0}) except Exception: log.error(f"get company id {comp_id} failed! ", exc_info=True) return jsonify({'code': 0})