-
Notifications
You must be signed in to change notification settings - Fork 2
/
server.py
72 lines (61 loc) · 1.85 KB
/
server.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#-*- coding:utf-8 -*-
from flask import Flask, jsonify, render_template, make_response, request
import os
from news import News
from conf import *
from fetch import Fetch
import time
import datetime
app = Flask(__name__, static_url_path='/assets')
def jsonResponse(data=None, type=None, extra_data=[]):
response = {}
status = 200
if type == None:
response['code'] = 0
response['result'] = data
if(extra_data):
[response.update(data) for data in extra_data]
elif type == '404':
status = 404
response = {"code": 404, "message": "Not Found"}
elif type == '500':
status = 500
response = {"code": 500, "message": "Server Eroor"}
response = jsonify(response)
response.status_code = status
return response
@app.route('/api/news')
def index():
news = News()
page = request.args.get('page')
if page is None:
page = 1
data = news.list(page=int(page), limit=20)
total = news.count()
if data is None:
return jsonResponse(type='404')
for i, item in enumerate(data):
data[i]['date'] = time.mktime(item['date'].timetuple())
return jsonResponse(data=data, extra_data=[{'total': total}])
@app.route('/api/news/<int:id>')
def news(id):
news = News(news_id=id)
data = news.get()
if data is None:
return jsonResponse(type='404')
return jsonResponse(data=data)
@app.route('/rss')
def rss():
news = News()
data = news.sort()
response = make_response(render_template('rss.xml', data=data, site_url=site_url))
response.headers['Content-Type'] = 'application/atom+xml; charset=utf-8'
return response
@app.route('/', defaults={'path': ''})
@app.route('/<path:path>')
def catch_all(path):
return render_template('index.html')
if __name__ == '__main__':
fetch = Fetch()
fetch.start()
app.run()