185 lines
6.6 KiB
Python
185 lines
6.6 KiB
Python
# -*- coding: utf-8 -*-
|
||
from __future__ import absolute_import
|
||
|
||
"""
|
||
广告接口
|
||
get （返回广告明细） params（gameid，localid）output（ad_title,ad_body,ad_image,ad_url）
|
||
post（新增广告定义） params（name,gameid,localid,start_time,end_time,ad_nums,ad_title,ad_body,ad_image,ad_url,ad_sort,status=unapproved,） output（1,0）
|
||
put（更新广告配置）params（name,gameid,localid,start_time,end_time,ad_nums,ad_title,ad_body,ad_image,ad_url,ad_sort）output（1,0）
|
||
delete（删除定义的广告）params（）output（1,0）
|
||
|
||
"""
|
||
|
||
from flask import Flask, jsonify
|
||
from flask_restful import reqparse, abort, Api, Resource
|
||
import logging
|
||
# from myredis.myredis import ad_redis, expire_time
|
||
from mysql.mmysql import MysqlBase
|
||
from config import mysql_promotion_config
|
||
import datetime
|
||
import pdb
|
||
|
||
log = logging.getLogger(__name__)
|
||
mydb = MysqlBase(**mysql_promotion_config)
|
||
parser = reqparse.RequestParser()
|
||
parser.add_argument('id')
|
||
parser.add_argument('gameid')
|
||
parser.add_argument('name')
|
||
parser.add_argument('locationid')
|
||
parser.add_argument('begin_time')
|
||
parser.add_argument('end_time')
|
||
parser.add_argument('ad_num')
|
||
parser.add_argument('ad_title')
|
||
parser.add_argument('ad_body')
|
||
parser.add_argument('ad_image')
|
||
parser.add_argument('ad_url')
|
||
parser.add_argument('ad_sort')
|
||
parser.add_argument('status')
|
||
parser.add_argument('companyid')
|
||
|
||
|
||
class Ad(Resource):
|
||
def __init__(self):
|
||
self.args = parser.parse_args()
|
||
|
||
def get(self):
|
||
status = self.args['status']
|
||
id = self.args['id']
|
||
companyid = self.args['companyid']
|
||
# if not gameid or not localid:
|
||
# # log.error(f"请输入必须的游戏ID和位置ID字段，当前获得是{gameid},{localid}")
|
||
# return jsonify({'code': 500, "message": f"请输入必须的游戏ID和位置ID字段，当前获得是{gameid},{localid},{status}"})
|
||
|
||
# now = datetime.datetime.today().strftime("%Y-%m-%d %H:%M:%S")
|
||
if not id:
|
||
if companyid:
|
||
sel_sql = f"select id,name,begin_time,end_time,ad_num,ad_title,ad_body,ad_image,ad_url,ad_sort," \
|
||
f"status,companyid,locationid,gameid from ad where companyid={companyid};"
|
||
elif status:
|
||
sel_sql = f"select id,name,begin_time,end_time,ad_num,ad_title,ad_body,ad_image,ad_url,ad_sort," \
|
||
f"status，companyid,locationid,gameid from ad where status={status};"
|
||
else:
|
||
sel_sql = f"select id,name,begin_time,end_time,ad_num,ad_title,ad_body,ad_image,ad_url,ad_sort," \
|
||
f"status,companyid,locationid,gameid from ad ;"
|
||
else:
|
||
sel_sql = f"select id,name,begin_time,end_time,ad_num,ad_title,ad_body,ad_image,ad_url,ad_sort," \
|
||
f"status,companyid,locationid,gameid from ad where id={id};"
|
||
data = mydb.query(sel_sql)
|
||
log.info(f"sql={sel_sql},data={data}")
|
||
if data:
|
||
all = []
|
||
ad_info = {}
|
||
try:
|
||
for line in data:
|
||
ad_info['id'], ad_info['name'], ad_info['begin_time'], ad_info['end_time'], ad_info['ad_num'], \
|
||
ad_info['ad_title'], ad_info['ad_body'], ad_info['ad_image'], ad_info['ad_url'], ad_info['ad_sort'], \
|
||
ad_info['status'], ad_info['companyid'], ad_info['locationid'], ad_info['gameid'] = line
|
||
all.append(ad_info)
|
||
return jsonify({'code': 200, 'message': all})
|
||
except Exception:
|
||
log.error("split data from mysql failed!", exc_info=True)
|
||
else:
|
||
return jsonify({
|
||
'code': 404, 'message': f'ad not found with id={id},status={status},companyid={companyid}!'
|
||
})
|
||
|
||
|
||
def post(self):
|
||
ad = {}
|
||
ad['id'] = self.get_adid()
|
||
ad['name'] = self.args['name']
|
||
ad['gameid'] = self.args['gameid']
|
||
ad['locationid'] = self.args['locationid']
|
||
ad['begin_time'] = self.args['begin_time'] or '1999-01-01'
|
||
ad['end_time'] = self.args['end_time'] or '3000-01-01'
|
||
ad['ad_num'] = self.args['ad_num'] or -1
|
||
ad['ad_title'] = self.args['ad_title']
|
||
ad['ad_body'] = self.args['ad_body']
|
||
ad['ad_image'] = self.args['ad_image']
|
||
ad['ad_url'] = self.args['ad_url']
|
||
ad['ad_sort'] = self.args['ad_sort'] or 0
|
||
ad['companyid'] = self.args['companyid']
|
||
# 检查必需有的字段
|
||
if not (
|
||
ad['name'] and ad['gameid'] and ad['locationid'] and ad['ad_title'] and ad['ad_body'] and ad['ad_image'] and
|
||
ad['ad_url'] and ad['companyid']):
|
||
return jsonify({'code': 500, 'message': '一些必填项未提供'})
|
||
|
||
# 检查该广告是否已存在
|
||
try:
|
||
check_sql = f"select id from ad where name='{ad['name']}' and gameid={ad['gameid']} and \
|
||
locationid={ad['locationid']} and companyid=ad['companyid']"
|
||
data = mydb.query(check_sql)
|
||
if data:
|
||
return jsonify({'code': 500, 'message': f"name={ad['name']} gameid={ad['gameid']} was in db!"})
|
||
except Exception:
|
||
log.error(f"check new id in db failed!", exc_info=True)
|
||
return jsonify({'code': 500, 'message': 'check data in db failed!'})
|
||
try:
|
||
mydb.insert('ad', ad)
|
||
return jsonify({'code': 200})
|
||
except Exception:
|
||
log.error("Insert ad to mysql failed!", exc_info=True)
|
||
return jsonify({'code': 500, 'message': f'insert {ad} failed! '})
|
||
|
||
def get_adid(self):
|
||
max_id = f"select id from ad order by id desc limit 1"
|
||
data = mydb.query(max_id)
|
||
try:
|
||
max = int(data[0][0]) + 1
|
||
log.info(f"max id was {max}!")
|
||
except Exception:
|
||
log.error(f"error with get ad id ", exc_info=True)
|
||
max = 1001
|
||
return max
|
||
|
||
def put(self):
|
||
try:
|
||
ad = {}
|
||
ad['id'] = self.args['id']
|
||
ad['name'] = self.args['name']
|
||
ad['gameid'] = self.args['gameid']
|
||
ad['locationid'] = self.args['locationid']
|
||
ad['begin_time'] = self.args['begin_time'] or '1999-01-01'
|
||
ad['end_time'] = self.args['end_time'] or '3000-01-01'
|
||
ad['ad_num'] = self.args['ad_num'] or -1
|
||
ad['ad_title'] = self.args['ad_title']
|
||
ad['ad_body'] = self.args['ad_body']
|
||
ad['ad_image'] = self.args['ad_image']
|
||
ad['ad_url'] = self.args['ad_url']
|
||
ad['ad_sort'] = self.args['ad_sort']
|
||
ad['status'] = self.args['status'] or 0
|
||
ad['companyid'] = self.args['companyid']
|
||
|
||
sel_sql = f"select name from ad where id={self.args['id']};"
|
||
data = mydb.query(sel_sql)
|
||
if data:
|
||
condition = f"id='{self.args['id']}'"
|
||
mydb.update("ad", ad, condition)
|
||
return jsonify({'code': 200})
|
||
else:
|
||
return jsonify({'code': 404, 'message': f"{self.args['id']} not found in mysql!"})
|
||
except Exception:
|
||
log.error("update values to ad mysql failed!", exc_info=True)
|
||
return jsonify({'code': 500})
|
||
|
||
|
||
def delete(self):
|
||
id = self.args['id'] or None
|
||
if not id:
|
||
return jsonify({'code': 404, 'message': f'{id} not found!'})
|
||
del_sql = f"delete from ad where id={id};"
|
||
try:
|
||
mydb.change(del_sql)
|
||
return jsonify({'code': 200, 'message': f'remove adid={id} success!'})
|
||
except Exception:
|
||
log.error("remove id from ad failed!", exc_info=True)
|
||
|
||
|
||
class Adlist(Resource):
|
||
def __init__(self):
|
||
self.args = parser.parse_args()
|
||
|
||
def get(self):
|
||
pass
|