def __authorize_by_code(request):
post_data = request.body.decode('utf-8')
post_data = json.loads(post_data)
code = post_data.get('code').strip()
app_id = post_data.get('appId').strip()
nickname = post_data.get('nickname').strip()
response_data = {}
if not code or not app_id:
response_data[
'message'] = 'authorized failed, need entire authorization data.'
response_data['code'] = ReturnCode.BROKEN_AUTHORIZED_DATA
return JsonResponse(data=response_data, safe=False)
data = c2s(app_id, code)
openid = data.get('openid')
print('get openid:', openid)
if not openid:
response = CommonResponseMixin.wrap_json_response(
code=ReturnCode.FAILED, message='auth failed')
return JsonResponse(data=response, safe=False)
request.session['open_id'] = openid
request.session['is_authorized'] = True
if not User.objects.filter(open_id=openid):
new_user = User(open_id=openid, nickname=nickname)
print('new_user:open_id:%s, nickname:%s' % (openid, nickname))
new_user.save()
response = CommonResponseMixin.wrap_json_response(code=ReturnCode.SUCCESS,
message='auth succeess')
return JsonResponse(data=response, safe=False)
def __authorize_by_code(request):
'''
使用wx.login得到的临时code到微信提供code2session接口授权
'''
post_data = request.body.decode('utf-8')
post_data = json.loads(post_data)
code = post_data['code'].strip()
app_id = post_data['appId'].strip()
nickname = post_data['nickname'].strip()
if not code or not app_id:
response['message'] = 'authorized failed,need entire authorization data'
response['code'] = ReturnCode.BROKEN_AUTHORIZED_DATA
return JsonResponse(data=response,safe=False)
data =c2s(app_id,code)
openid = data['openid']
if not openid:
response = CommonResponseMixin.wrap_json_response(code=ReturnCode.FAILED,message='auth failed')
return JsonResponse(data=response,safe=False)
request.session['open_id'] = openid
request.session['is_authorized'] = True
if not User.objects.filter(open_id=openid):
new_user = User(open_id=openid,nickname=nickname)
new_user.save()
response = CommonResponseMixin.wrap_json_response(code=ReturnCode.SUCCESS,message='auth success')
return JsonResponse(data=response,safe=False)
def __authorization_by_code(request):
data = request.body.decode("utf-8")
data = json.loads(data)
code = data.get('code').strip()
appId = data.get('appId').strip()
nickName = data.get('nickName').strip()
response = {}
if not code or not appId:
response['message'] = '登陆失败，需要完整数据'
response['code'] = ReturnCode.BROKEN_AUTHORIZED_DATA
return JsonResponse(data=response, safe=False)
data = c2s(appId, code)
openid = data.get('openid')
if not openid:
response = CommonResponseMixin.wrap_json_response(code=ReturnCode.FAILED, message="auth failed.")
return JsonResponse(data=response)
request.session['open_id'] = openid
request.session['is_authorized'] = True
if not Yonghu.objects.filter(open_id=openid):
user = Yonghu(open_id=openid, nickname=nickName)
user.save()
print("new user: openid: %s, nickName: %s" % (openid, nickName))
response = CommonResponseMixin.wrap_json_response(code=ReturnCode.SUCCESS, message='auth success')
return JsonResponse(data=response, safe=False)
def get_host_disk_guage(request):
hostname = request.GET.get('host')
if hostname == 'All':
hostname = ''
elif not hostname:
return JsonResponse(data=CommonResponseMixin.wrap_json_response(code=ReturnCode.WRONG_PARAMS))
data = promVector.host_disk_guage(hostname)
response_data = CommonResponseMixin.wrap_json_response(data=data)
return JsonResponse(data=response_data, safe=False)
def get_taskId_list(request):
queryType = request.GET.get('type')
if queryType == 'clamavfileid' or queryType == 'multiclamdid':
value = cache.get(queryType)
if not value:
value = 'None' # 没值要附值，不然前端没有二层data字段
response_data = CommonResponseMixin.wrap_json_response(data=value)
return JsonResponse(data=response_data, safe=False)
response_data = CommonResponseMixin.wrap_json_response(code=ReturnCode.WRONG_PARAMS)
return JsonResponse(data=response_data, safe=False)
def image_text(request):
if request.method == 'GET':
md5 = request.GET.get('md5')
imgfile = os.path.join(settings.IMAGES_DIR, md5 + '.jpg')
if not os.path.exists(imgfile):
return CommonResponseMixin.wrap_json_response(
code=ReturnCode.RESOURCES_NOT_EXISTS)
else:
response_data = {}
response_data["name"] = md5 + '.jpg'
response_data["url"] = "/service/image?md5=%s" % (md5)
data = CommonResponseMixin.wrap_json_response(data=response_data)
return JsonResponse(data=data, safe=False)
def post(self, request):
if not already_authorized(request):
response = self.wrap_json_response(code=ReturnCode.UNAUTHORIZED)
return JsonResponse(response, safe=False)
open_id = request.session.get('open_id')
user = Tbluserinfo.objects.get(open_id=open_id)
# got str object
received_body = request.body.decode('utf-8')
received_body = eval(received_body)
cities = received_body.get('city')
stocks = received_body.get('stock')
constellations = received_body.get('constellation')
if cities == None: cities = []
if stocks == None: stocks = []
if constellations == None: constellations = []
user.focus_cities = json.dumps(cities)
user.focus_constellations = json.dumps(constellations)
user.focus_stocks = json.dumps(stocks)
user.save()
message = 'modify user info success.'
response = CommonResponseMixin.wrap_json_response(
code=ReturnCode.SUCCESS, message=message)
return JsonResponse(response, safe=False)
def isExist(request):
if request.method == "GET":
usericode = request.GET.get('usericode')
try:
Tbluserinfo.objects.get(usericode=usericode)
message = 'already exist a user.'
response = CommonResponseMixin.wrap_json_response(
code=ReturnCode.FAILED, message=message)
return JsonResponse(response, safe=False)
except:
message = 'can create new user.'
response = CommonResponseMixin.wrap_json_response(
code=ReturnCode.SUCCESS, message=message)
return JsonResponse(response, safe=False)
else:
return HttpResponse("error method")
def delete(self, request):
ID = request.session.get('ID')
obj = Tbluserinfo.objects.filter(id=ID).update(delflag=1)
message = 'delete user info success.'
response = CommonResponseMixin.wrap_json_response(
code=ReturnCode.SUCCESS, message=message)
return JsonResponse(response, safe=False)
def all_category(request):
categorys = Category.objects.all()
data = []
for category in categorys:
data.append(category.name)
response_data = CommonResponseMixin.wrap_json_response(data=data)
return JsonResponse(data=response_data, safe=False)
def logout(request):
# print(request.session)
# response = CommonResponseMixin.wrap_json_response(message="GG")
# return JsonResponse(response, safe=False)
request.session.clear()
response = CommonResponseMixin.wrap_json_response(code=ReturnCode.SUCCESS)
return JsonResponse(data=response, safe=False)
def get_author(request):
post_data = request.body.decode('utf-8')
post_data = json.loads(post_data)
code = post_data.get('code').strip()
appid = post_data.get('appid').strip()
nickName = post_data.get('nickName').strip()
response = {}
if not code or not appid:
response['message'] = 'authorize failed'
response['code'] = ReturnCode.BROKEN_AUTHORIZED_DATA
return JsonResponse(data=response, safe=False)
data = auth.c2s(appid, code)
openid = data.get('openid')
if not openid:
response['message'] = 'authorize failed'
response['code'] = ReturnCode.BROKEN_AUTHORIZED_DATA
return JsonResponse(data=response, safe=False)
request.session['open_id'] = openid
request.session['is_authorized'] = True
if not User.objects.filter(openid=openid):
user = User(openid=openid, nickname=nickName)
user.save()
response = CommonResponseMixin.wrap_json_response(code=ReturnCode.SUCCESS,
message='success auth')
return JsonResponse(data=response, safe=False)
def get_staus(request):
if already_authorized(request):
data = {"is_authorized": 1}
else:
data = {"is_authorized": 0}
response = CommonResponseMixin.wrap_json_response(data=data, code=ReturnCode.SUCCESS)
return JsonResponse(data=response, safe=False)
def get_article(request):
article_id = request.GET.get('article_id')
article = Article.objects.get(article_id=article_id)
# print('article_id', article_id)
article.pageview = article.pageview + 1 # 阅读量
article.save()
data = dict()
data['content'] = []
data['title'] = article.title
data['author'] = article.author.nickname
data['pageview'] = article.pageview
brief_content = article.brief_content.split(',')
bodys = article.content.split('###') # 按章节分的列表
for index, body in enumerate(bodys):
tempDuanluo = []
tempDict = dict()
chapter = body.replace('\r', '')
# chapter = body
paragraphs = chapter.split('\n') # 按段落再分,可能会出现空段落
for paragraph in paragraphs:
if paragraph: # 空段落则跳过
tempDuanluo.append(paragraph)
tempDict['brief'] = brief_content[index]
tempDict['body'] = tempDuanluo
data['content'].append(tempDict)
response_data = CommonResponseMixin.wrap_json_response(data=data)
return JsonResponse(data=response_data, safe=False)
def post(self, request):
if not already_authorized(request):
response = self.wrap_json_response(code=ReturnCode.UNAUTHORIZED)
return JsonResponse(response, safe=False)
open_id = request.session.get('open_id')
user = UserProfile.objects.get(open_id=open_id)
# got str object
received_body = request.body.decode('utf-8')
received_body = eval(received_body)
nickname = received_body.get('nickname')
birthday = received_body.get('birthday')
gender = received_body.get('gender')
address = received_body.get('address')
signature = received_body.get('signature')
user.nickname = nickname
user.birthday = birthday
user.gender = gender
user.address = address
user.signature = signature
user.save()
message = 'modify user info success.'
response = CommonResponseMixin.wrap_json_response(
code=ReturnCode.SUCCESS, message=message)
return JsonResponse(response, safe=False)
def joke(request):
global all_jokes
if not all_jokes:
all_jokes = json.load(open(os.path.join(settings.BASE_DIR, 'jokes.json'), 'r'))
limits = 10
sample_jokes = random.sample(all_jokes, limits)
response_data = CommonResponseMixin.wrap_json_response(data=sample_jokes)
return JsonResponse(data=response_data,safe=False)
def stock(request):
data = []
for stock in popular_stocks:
result = thirdparty.juhe.stock(stock['market'], stock['code'])
data.append(result)
response = CommonResponseMixin.wrap_json_response(data=data,
code=ReturnCode.SUCCESS)
return JsonResponse(data=response, safe=False)
def constellation(request):
data = []
for c in constellations:
result = thirdparty.juhe.constellation(c)
data.append(result)
response = CommonResponseMixin.wrap_json_response(data=data,
code=ReturnCode.SUCCESS)
return JsonResponse(response, safe=False)
def news(request):
data = cache.get('news')
if not data:
data = parse_sohu()
cache.set('news', data, 60 * 60) # 1h
response_data = CommonResponseMixin.wrap_json_response(data=data)
return JsonResponse(data=response_data, safe=False)
def get_status(request):
print('call get_status function...')
if already_authorized(request):
data = {'is_authorized': 1}
else:
data = {'is_authorized': 0}
response = CommonResponseMixin.wrap_json_response(data=data, code=ReturnCode.SUCCESS)
return JsonResponse(data=response, safe=False)
def getDutyList(request):
if (request.method == "GET"):
data = models.Tbldutyinfo.objects.all().values()
print(list(data))
response = CommonResponseMixin.wrap_json_response(
code=ReturnCode.SUCCESS, data=list(data))
return JsonResponse(response, safe=False)
else:
return HttpResponse("error")
def all_record(request):
query_set = Ldns.objects.all()
print('query_set', query_set)
all_dns = []
for dns in query_set:
all_dns.append(dns.to_dict())
print(all_dns)
response_data = CommonResponseMixin.wrap_json_response(data=all_dns)
return JsonResponse(data=response_data, safe=False)
def joke(request):
global joke_cache
if not joke_cache:
joke_cache = json.load(open(os.path.join(settings.BASE_DIR, 'jokes.json'), 'r'))
# 读缓存
all_jokes = joke_cache
limit = 10
sample_jokes = random.sample(all_jokes, limit)
response = CommonResponseMixin.wrap_json_response(data=sample_jokes, code=ReturnCode.SUCCESS)
return JsonResponse(response, safe=False)
def constellation(request):
if already_authorized(request):
user = get_user(request)
constellations = json.loads(user.focus_constellations)
else:
constellations = ['', '', '']
# print(constellations)
response = CommonResponseMixin.wrap_json_response(data=constellations,
code=ReturnCode.SUCCESS)
return JsonResponse(data=response, safe=False)
def stock(request):
if already_authorized(request):
user = get_user(request)
stock = json.loads(user.focus_stocks)
else:
stock = ['', '']
# print(stock)
response = CommonResponseMixin.wrap_json_response(data=stock,
code=ReturnCode.SUCCESS)
return JsonResponse(data=response, safe=False)
def get(self, request):
# if not already_authorized(request):
# response = self.wrap_json_response(code=ReturnCode.UNAUTHORIZED)
# return JsonResponse(response, safe=False)
ID = request.session.get('ID')
data = Tbluserinfo.objects.filter(id=ID).values()
response = CommonResponseMixin.wrap_json_response(
code=ReturnCode.SUCCESS, data=list(data))
return JsonResponse(response, safe=False)
def cropId(request):
if already_authorized(request):
user = get_user(request)
cropIds = json.loads(user.focus_cropIds)
else:
cropIds = all_cropIds
response = CommonResponseMixin.wrap_json_response(data=cropIds,
code=ReturnCode.SUCCESS)
return JsonResponse(data=response, safe=False)
def get_containers_cpu(request):
hostname = request.GET.get('hostname') # 啥都不传默认集群和10分钟的区间
if hostname == 'All':
hostname = ''
interval = request.GET.get('interval') # 只能传秒为单位的值！！！
if interval:
response_data = cpuContainers(hostname=hostname,
time_interval=int(interval))
else:
response_data = cpuContainers(hostname=hostname)
response_data = CommonResponseMixin.wrap_json_response(data=response_data)
return JsonResponse(data=response_data, safe=False)
def get_menu(request):
# global_apps_data = init_app_data()
# published_apps_data = global_apps_data.get('published')
# response = CommonResponseMixin.wrap_json_response(data=published_apps_data, code=ReturnCode.SUCCESS)
# return JsonResponse(data=response, safe=False)
query_set = App.objects.all()
all_app = []
for app in query_set:
all_app.append(app.to_dict())
print(all_app)
response = CommonResponseMixin.wrap_json_response(data=all_app)
return JsonResponse(data=response, safe=False)
def constellation(request):
data = []
if already_authorized(request):
user = get_user(request)
constellations = json.loads(user.focus_constellations)
else:
constellations = all_constellations
for c in constellations:
result = juhe.constellation(c)
data.append(result)
response_data = CommonResponseMixin.wrap_json_response(data=data, code=ReturnCode.SUCCESS)
return JsonResponse(data=response_data, safe=False)
