def test_change(self): conn = Database.conn Database.change(mysql_db) assert Database.conn is conn Database.config(db=mysql_db) Database.execute('insert into user set user.name="test"') assert Database.conn_is_up() assert Database.conn is not conn
def channel_set(request, param): if request.method == "GET": query_filter = {} try: query_filter['channel'] = request.GET.get('channel', '') query_filter['start_time'] = request.GET.get('start_time', '') query_filter['end_time'] = request.GET.get('end_time', '') except Exception as e: raise res = A_channel_set.get_list(query_filter) print(type(res)) option = {'is_public': global_conf.is_public, } res_1 = utils.prepare_table_data(res, option) return JsonResponse(res_1 if res_1 else res, safe = False) elif request.method == "POST": try: keys = ['channel', 'parent_id', 'weight', 'remark', 'channel_type', 'is_public'] par = utils.get_post_parameter(request, keys) par['admin'] = request.user.username par['status'] = 1 except Exception as e: print(traceback.format_exc()) return JsonResponse({"status": 1, "message":"参数错误"}) a_channel_set = A_channel_set() a_channel_set = utils.model_set(a_channel_set, par) try: a_channel_set.save() except Exception as e: return JsonResponse({"status": 1, "message":"渠道名相同"}) return JsonResponse({"status": 0, "message":"添加成功"}) elif request.method == "PUT": id = int(param) try: keys = ['channel', 'parent_id', 'weight', 'remark', 'channel_type', 'is_public'] par = utils.get_post_parameter(request, keys) par['admin'] = request.user.username except Exception as e: print(traceback.format_exc()) return JsonResponse({"status": 1, "message":"参数错误"}) a_channel_set = A_channel_set.where(id=id).select().execute.one() a_channel_set = utils.model_set(a_channel_set, par) try: a_channel_set.save() except Exception as e: return JsonResponse({"status": 1, "message":"渠道名相同"}) return JsonResponse({"status": 0, "message":"编辑成功"}) elif request.method == "DELETE": id = int(param) sql = "UPDATE a_channel_set SET status = 0 WHERE id = %s" % (id) Database.execute(sql) return JsonResponse({"status": 0, "message":"删除成功"})
def test_execute(self): cursor = Database.execute('insert into user set user.name="test"') assert cursor