added simple server & data builder

Author: bcho

builder.py

@@ -0,0 +1,43 @@
+#coding: utf-8
+
+'''
+    builder.py
+    ~~~~~~~~~~
+
+    Build data.
+'''
+
+import db
+
+
+def pick(size, start_id=None):
+    if start_id:
+        cur = db.get_page_by_id(start_id)
+    else:
+        cur = db.get_random_page()
+
+    dataset = [cur]
+    while len(dataset) < size:
+        for i in cur.relatives:
+            if i not in dataset:
+                dataset.append(i)
+        if dataset.index(cur) + 1 >= len(dataset):
+            while cur in dataset:
+                cur = db.get_random_page()
+            dataset.append(cur)
+        else:
+            cur = dataset[dataset.index(cur) + 1]
+    while len(dataset) > size:
+        dataset.pop()
+
+    return dataset
+
+
+def build(size, start_id=None):
+    raw = pick(size, start_id)
+
+    dataset = {}
+    for i in raw:
+        dataset[i.id] = i.dict
+
+    return dataset

config.py

@@ -23,3 +23,6 @@
 
 #: PageRank settings
 pr_iterate_times = 100
+
+#: server settings
+DEBUG = True

db.py

@@ -1,5 +1,7 @@
 #coding: utf-8
 
+from random import randint
+
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, scoped_session
 from sqlalchemy.sql import and_
@@ -35,10 +37,22 @@ def _get_one(model, cond):
     return ret[0] if ret else None
 
 
+def pages_count():
+    return session.query(Page).count()
+
+
+def get_page_by_id(id):
+    return _get_one(Page, (Page.id == id))
+
+
 def get_page(name, section):
     return _get_one(Page, and_(Page.name == name, Page.section == section))
 
 
+def get_random_page():
+    return get_page_by_id(randint(1, pages_count()))
+
+
 def add_page(name, section, relatives=None):
     #: replace the empty section with 3
     section = section or '3'

manage.py

@@ -1,7 +1,29 @@
 #coding: utf-8
 
+import sys
+
 import db
+import server
+
+
+def _main():
+    command = 'server'
+    if len(sys.argv) > 1:
+        command = sys.argv[1]
+
+    if command == 'server':
+        server.app.run()
+    elif command == 'rating':
+        if sys.argv[2] == 'init':
+            db.init_rating()
+        elif sys.argv[2] == 'update':
+            db.rating()
+    elif command == 'db':
+        if sys.argv[2] == 'drop':
+            db.drop_all()
+        elif sys.argv[2] == 'create':
+            db.create_all()
 
 
 if __name__ == '__main__':
-    db.rating()
+    _main()

models.py

@@ -1,5 +1,7 @@
 #coding: utf-8
 
+import json
+
 from sqlalchemy import Column, Integer, String, Float
 from sqlalchemy import ForeignKey, Table
 from sqlalchemy.orm import relationship
@@ -40,3 +42,20 @@ def __repr__(self):
     @property
     def relatives(self):
         return list(set(self.fromme + self.tome))
+
+    @property
+    def dict(self):
+
+        def _relatives(rel):
+            return [i.id for i in rel]
+
+        return {
+            'fromme': _relatives(self.fromme),
+            'tome': _relatives(self.tome),
+            'name': self.name,
+            'id': self.id,
+            'rating': self.rating,
+        }
+
+    def to_json(self):
+        return json.dumps(self.dict)

requirements.txt

@@ -1,2 +1,5 @@
+Flask==0.9
+Jinja2==2.6
 SQLAlchemy==0.7.9
+Werkzeug==0.8.3
 wsgiref==0.1.2

server.py

@@ -0,0 +1,17 @@
+#coding: utf-8
+
+from flask import Flask, request, jsonify
+
+import builder
+
+#: init app
+app = Flask(__name__)
+app.config.from_pyfile('config.py')
+
+
+@app.route('/pages.json')
+def get():
+    limit = int(request.args.get('limit', 20))
+    start_id = request.args.get('start_id', None)
+    start_id = int(start_id) if start_id else None
+    return jsonify(builder.build(limit, start_id))