-
Notifications
You must be signed in to change notification settings - Fork 28
Expand file tree
/
Copy pathapi.py
More file actions
executable file
·120 lines (99 loc) · 2.93 KB
/
api.py
File metadata and controls
executable file
·120 lines (99 loc) · 2.93 KB
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
#!/usr/bin/env python
# coding: utf8
from sqlalchemy.pool import QueuePool
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, exc
import models
import logger
engine = None
maker = None
BASE = models.BASE
log =None
def configure_db():
global engine, session, log
if not engine:
poolclass=QueuePool,
echo_pool=True,
pool_recycle=3600)
register_models()
log = logger.get_logger()
log.info("db inited.")
def clear_db():
global engine
assert engine
for table in reversed(BASE.metadata.sorted_tables):
engine.execute(table.delete())
def register_models():
"""Register Models and create properties"""
global engine
assert engine
BASE.metadata.create_all(engine)
def unregister_models():
"""Un register Models, useful clearing out data before testing"""
global engine
assert engine
BASE.metadata.drop_all(engine)
def get_session(autocommit=True, expire_on_commit=False):
"""Helper method to grab session"""
global maker, engine
if not maker:
assert engine
maker = sessionmaker(bind=engine,
autocommit=autocommit,
expire_on_commit=expire_on_commit)
return maker()
def news_create(title, url, site, **kwargs):
session = get_session()
with session.begin():
news = models.News(title, url, site)
for key in kwargs.keys():
news[key] = kwargs[key]
session.add(news)
session.flush()
return news
def news_list(page_size, page=0, filters=None):
log.info("Listing news ...")
session = get_session()
query = session.query(models.News)
if filters:
query = query.filter(filters)
query.limit(page_size)
query = query.limit(page_size)
query = query.offset(page * page_size)
return query.all()
def news_get(id):
session = get_session()
return session.query(models.News).filter_by(id=id).one()
def news_update(id, **kwargs):
session = get_session()
news = news_get(id)
if not news:
return None
for key in kwargs.keys():
news[key] = kwargs[key]
session.merge(news)
session.flush()
return news
def news_destroy(id):
session = get_session()
try:
news = session.query(models.News).filter_by(id=id).one()
session.delete(news)
session.flush()
except:
raise Exception("News %d not found" % id)
if __name__ == '__main__':
configure_db()
import time
while True:
news = news_list(20)
for n in news:
print n.title, n.url
print news_get(2534).title
news = news_create("test tile", "http://www.test.com", "test")
print news_get(news.id)
news = news_update(news.id, abstract="This is a test")
print news
news_destroy(news.id)
time.sleep(0.1)