From 5d0d0ac81677459bacc8bcb80fcbcc6389b67958 Mon Sep 17 00:00:00 2001 From: pengtao Date: Mon, 8 Jul 2019 19:53:49 +0800 Subject: [PATCH] add ad --- handler/company.py | 32 +++++++++++++++++++++----------- 1 file changed, 21 insertions(+), 11 deletions(-) diff --git a/handler/company.py b/handler/company.py index 4553007..113b768 100644 --- a/handler/company.py +++ b/handler/company.py @@ -80,11 +80,11 @@ class Company(Resource): company['appid'] = args['appid'] company['appkey'] = args['appkey'] # insert to mysql - #update_sql = f"update company set id='{company['id']}',name='{company['name']}',contact='{company[ + # update_sql = f"update company set id='{company['id']}',name='{company['name']}',contact='{company[ # 'contact']}',appid='{company['appid']}',appkey='{company['appkey']}' where id='{comp_id}';" - condition=f"id='{comp_id}'" - #log.info(f"update sql was {update_sql}") - mydb.update("company", company,condition) + condition = f"id='{comp_id}'" + # log.info(f"update sql was {update_sql}") + mydb.update("company", company, condition) # update redsi company_redis.hmset(company['id'], company) @@ -102,11 +102,21 @@ class CompanyList(Resource): return jsonify({'code': 1, 'body': company_redis.lrang("ALL_Company", 0, -1)}) def post(self): - args = parser.parse_args() - log.info(f"args was {args}") - try: - company_redis.rpush("ALL_Company", args['company']) - except Exception: - log.error("Insert values to redis failed,args was {}!".format(args), exc_info=True) + companys = self.get_company() + if companys: + return jsonify({'code': 1, 'body': companys}) + else: return jsonify({'code': 0}) - return jsonify({'code': 1}) + + + def get_company(self): + sel_sql = "select id from company where 1;" + data = mydb.query(sel_sql) + company_list = [] + if data: + for item in data: + try: + company_list.append(item[0]) + except Exception: + log.error("split data from db failed!", exc_info=True) + return company_list