init commit

Author: misakar

Diff for: .gitignore

@@ -0,0 +1,2 @@
+**/*.pyc
+venv/

Diff for: config.py

@@ -0,0 +1,18 @@
+# coding: utf-8
+
+import os
+from datetime import timedelta
+
+
+class Config(object):
+
+    XNM = 2016
+    XQM = 3
+
+
+config = {
+    'develop': Config,
+    'test': Config,
+
+    'default': Config
+}

Diff for: manage.py

@@ -0,0 +1,16 @@
+# coding: utf-8
+
+import sys
+from tables import app
+from gevent import monkey
+monkey.patch_all()
+
+
+if __name__ == '__main__':
+    if len(sys.argv) == 3:
+        path = sys.argv[2]
+        host, port = path.split(':')
+    else:
+        host = "127.0.0.1"; port = "5060";
+    if len(sys.argv) > 1 and sys.argv[1] == 'runserver':
+        app.run(debug=True, host=host, port=int(port))

Diff for: readme.md

@@ -0,0 +1,3 @@
+# restccnu_table
+
+table service for restccnu

Diff for: requirements.txt

@@ -0,0 +1,18 @@
+beautifulsoup4==4.5.1
+bs4==0.0.1
+click==6.6
+Flask==0.11.1
+futures==3.0.5
+gevent==1.1.2
+greenlet==0.4.10
+itsdangerous==0.24
+Jinja2==2.8
+MarkupSafe==0.23
+mongokit==0.9.1.1
+msgpack-python==0.4.8
+neovim==0.1.9
+pymongo==2.8
+requests==2.11.1
+six==1.10.0
+trollius==2.1
+Werkzeug==0.11.11

Diff for: tables/__init__.py

@@ -0,0 +1,17 @@
+# coding: utf-8
+
+from flask import Flask, render_template, request
+from config import config
+
+
+def create_app(config_name='default'):
+    app = Flask(__name__)
+    app.config.from_object(config[config_name])
+
+    from apis import api
+    app.register_blueprint(api, url_prefix='/api')
+
+    return app
+
+
+app = create_app()

Diff for: tables/apis/__init__.py

@@ -0,0 +1,67 @@
+# coding: utf-8
+
+from flask import Blueprint, jsonify
+
+
+api = Blueprint('api', __name__)
+
+
+@api.route('/')
+def api_index():
+    return jsonify({
+        'meta': {
+            'project': 'restccnu',
+            'version': 'v1',
+            'source code': 'https://github.com/Muxi-Studio/restccnu',
+            'author': ['@neo1218', '@kasheemlew'],
+        },
+        'apis': {
+            'Information Portal Login': '/api/info/login/',
+            'Library': [
+                {'library login': '/api/lib/login/'},
+                {'book search': '/api/lib/search/?keyword=xxx&page=n'},
+                {'book detail': '/api/lib/?id=xxxxxx&book=xxxx&author=xxxxx'},
+                {'my library': '/api/lib/me/'},
+            ],
+            'Class Schedule': [
+                {'class schedule query': '/api/table/?xnm=n&xqm=n'},
+                {'add personal class': '/api/table/'},
+                {'delete personal class': '/api/table/id/'}
+            ],
+            'Grade Query': {
+                'query': '/api/grade/search/?xnm=n&xqm=n',
+                'detail query': '/api/grade/detail/search/?xnm=x&xqm=x&course=x&jxb_id=x'
+            },
+            'Info API': '/api/info/',
+            'Electric bill API': '/api/ele/',
+            'Apartment info API': '/api/apartment/',
+            'Static Management': [
+                {'banner api': '/api/banner/'},
+                {'calendar api': '/api/calendar'},
+                {'start api': '/api/start/'}
+            ],
+            'Version Management': {
+                'app version management': [
+                    {'all ccnubox version': '/api/app/'},
+                    {'add a new version': '/api/app/'},
+                    {'delete a specific version': '/api/app/'},
+                    {'ccnubox latest version': '/api/app/latest/'}
+                ],
+                'patch version management': [
+                    {'all ccnubox patches version': '/api/patch/'},
+                    {'add a new patch version': '/api/patch/'},
+                    {'delete a specific patch version': '/api/patch/'},
+                    {'latest ccnubox patch version': '/api/patch/latest/'}
+                ],
+            },
+            'muxistudio products': [
+                {'muxistudio products': '/api/product/'},
+                {'add a product': '/api/product/'},
+                {'delete a product': '/api/product/'},
+                {'update a product': '/api/product/'},
+            ],
+        },
+    }), 200
+
+
+from . import table, login

Diff for: tables/apis/decorators.py

@@ -0,0 +1,62 @@
+# coding: utf-8
+
+import os
+import base64
+import functools
+import json
+from flask import make_response, jsonify, request, abort
+from tables.spiders.login import info_login
+from tables.errors import ForbiddenError
+
+
+def tojson(f):
+    @functools.wraps(f)
+    def decorator(*args, **kwargs):
+        rv = f(*args, **kwargs)
+        status_or_headers = None
+        headers = None
+        if isinstance(rv, tuple):
+            rv, status_or_headers, headers = rv + (None, ) * (3 - len(rv))
+        if isinstance(status_or_headers, (dict, list)):
+            headers, status_or_headers = status_or_headers, None
+
+        rv = json.dumps(rv, indent=1, ensure_ascii=False)
+        rv = make_response(rv)
+        if status_or_headers is not None:
+            rv.status_code = status_or_headers
+        if headers is not None:
+            rv.headers.extend(headers)
+        return rv
+    return decorator
+
+
+def require_info_login(f):
+    @functools.wraps(f)
+    def decorator(*args, **kwargs):
+        try:
+            s, sid = info_login()
+        except ForbiddenError as e:
+            return jsonify({}), e.status_code
+        else:
+            rv = f(s, sid, *args, **kwargs)
+            return rv
+    return decorator
+
+
+def admin_required(f):
+    @functools.wraps(f)
+    def decorator(*args, **kwargs):
+        # http basic auth: Basic base64(email:password)
+        token_header = request.headers.get('authorization', None)
+        if token_header:
+            email_pass = base64.b64decode(token_header[6:])
+            email = email_pass.split(':')[0]
+            password = email_pass.split(':')[1]
+            if email == os.getenv('QINIU_EMAIL') and \
+               password == os.getenv('QINIU_PASS'):
+                return f(*args, **kwargs)
+            else:
+                return jsonify({'msg': 'unauthorized'}), 401
+        else:
+            return jsonify({'msg': 'forbidden'}), 403
+    return decorator

Diff for: tables/apis/login.py

@@ -0,0 +1,11 @@
+# coding: utf-8
+
+from . import api
+from flask import jsonify
+from .decorators import require_info_login
+
+
+@api.route('/info/login/')
+@require_info_login
+def api_info_login(s, sid):
+    return jsonify({}), 200

Diff for: tables/apis/table.py

@@ -0,0 +1,83 @@
+# coding: utf-8
+
+import base64
+import json
+from flask import request, jsonify, current_app
+from .decorators import require_info_login, tojson
+from . import api
+from ..spiders.table import get_table
+from ..spiders.login import info_login
+from tables.errors import ForbiddenError, NotfoundError
+from tables.models import connection, User, _zero  # 占位课程(id=0)
+
+
+@api.route('/table/')
+@require_info_login
+@tojson
+def api_get_table(s, sid):
+    # xnm = request.args.get('xnm')
+    # xqm = request.args.get('xqm')
+    xnm = current_app.config['XNM']
+    xqm = current_app.config['XQM']
+
+    rv = get_table(s, sid, xnm, xqm)
+    user = connection.User.find_one({'sid': sid})
+    if user is None:
+        u = connection.User()
+        u['sid'] = sid; u['table'] = [_zero]
+        u.save()
+    user = connection.User.find_one({'sid': sid})
+
+    infocourse_dict = []
+    for index, course in enumerate(rv):
+        course.update({'color': index - 4*(index//4)})
+        infocourse_dict.append(course)
+
+    return user['table'] + infocourse_dict
+
+
+@api.route('/table/', methods=['POST'])
+@require_info_login
+def api_add_table(s, sid):
+    if request.method == 'POST':
+        user = connection.User.find_one({'sid': sid})
+        if user is None:
+            return jsonify({}), 404
+
+        course = request.get_json().get('course')
+        teacher = request.get_json().get('teacher')
+        weeks = request.get_json().get('weeks')
+        day = request.get_json().get('day')
+        start = request.get_json().get('start')
+        during = request.get_json().get('during')
+        place = request.get_json().get('place')
+        remind = request.get_json().get('remind')
+        id = request.get_json().get('id')
+        new_json = {'course': course, 'teacher': teacher, 'weeks': weeks,
+                    'day': day, 'start': start, 'during': during,
+                    'place': place, 'remind': remind, 'id': id, 'color': 0}
+        table = user['table']
+        for item in table:
+            if item.get('id') == str(id):
+                return jsonify({'bad request':
+                                'id already exist in database'}), 400
+        table.append(new_json)
+        user['table'] = table;
+        user.save()
+
+        return jsonify({}), 201
+
+
+@api.route('/table/<int:id>/', methods=['DELETE'])
+@require_info_login
+def api_delete_table(s, sid, id):
+    if request.method == 'DELETE':
+        user = connection.User.find_one({'sid': sid})
+        if user is None:
+            return jsonify({}), 404
+        table = user['table']
+        for i, item in enumerate(table):
+            if item.get('id') == str(id):
+                del table[i]
+        user.save()
+        return jsonify({}), 200

Diff for: tables/errors.py

@@ -0,0 +1,17 @@
+# coding: utf-8
+
+
+class ForbiddenError(Exception):
+    def __init__(self):
+        self.status_code = 403
+
+    def __repr__(self):
+        return "{'msg': 'forbidden'}"
+
+
+class NotfoundError(Exception):
+    def __init__(self):
+        self.status_code = 404
+
+    def __repr__(self):
+        return "{'msg': 'notfound'}"

Diff for: tables/models/__init__.py

@@ -0,0 +1,27 @@
+# coding: utf-8
+from .mongodoc import User, Dormitory
+from mongokit import Connection
+
+
+# config
+MONGODB_HOST = '123.56.41.13'
+MONGODB_PORT = 27020
+
+#  使用mongodb进行课表数据存储
+connection = Connection(host=MONGODB_HOST, port=MONGODB_PORT)
+connection.register([User])
+connection.register([Dormitory])
+
+# _zero: 占位课程, id=0
+## mongodb😓 的特性, 只有数据写入的时候创建数据库
+_zero = {
+    "id": "0",
+    "course": "re:从零开始的异世界生活",
+    "teacher": "neo1218",
+    "weeks": "1",
+    "day": "2",
+    "start": "3",
+    "during": "4",
+    "place": "5",
+    "remind": False
+}

Diff for: tables/models/mongodoc.py

@@ -0,0 +1,34 @@
+# coding: utf-8
+from mongokit import Document
+
+
+class User(Document):
+    """
+    存储用户对应的课表json数据(全部)
+    mongodb文档结构: db->collection->data
+    """
+    __collection__ = 'users'
+    __database__ = 'userdb'
+    structure = {
+            'sid': basestring,
+            'table': list
+    }
+    required_fields = ['sid', 'table']
+
+    def __repr__(self):
+        return '<Mongo User>'
+
+
+class Dormitory(Document):
+    """
+    存储 _meter_index 字典
+    """
+    __collection__ = 'dormitories'
+    __database__ = 'dordb'
+    structure = {
+            'meter': dict
+    }
+    required_fields = ['meter']
+
+    def __repr__(self):
+        return '<Mongo Dormitory>'