import os.path
-import re
+import shutil,re
import tornado.escape
import tornado.web
-import pymongo
+import tornado.httpserver
+import tornado.ioloop
+import tornado.options
+from tornado.options import define,options
+from tinydb import TinyDB,Query,where
+from tinydb.operations import delete
from datetime import datetime
+import json
+
+define('port',default=8000,help='run on the given port',type=int)
class BaseHandler(tornado.web.RequestHandler):
def get_current_user(self):
class IndexHandler(BaseHandler):
def get(self,dbname,page='0'):
- params = self.application.db['params'].find_one()
+ params = self.application.db.get(where('kinds') == 'conf')
if params['mentenance'] == True:
self.render('mentenance.htm',title=params['title'],db=dbname)
if self.application.collection(dbname) == False:
if self.current_user == b'admin':
- self.application.db[dbname]
+ self.application.db.table(dbname)
+ else:
+ raise tornado.web.HTTPError(404)
+ return
+ key = self.get_argument('key','')
+ if key:
+ table = self.application.db.table(dbname)
+ rec = table.get(where('number') == int(key))
+ if rec:
+ self.render('article.htm',record=rec)
+ return
else:
raise tornado.web.HTTPError(404)
return
i = params['count']
na = tornado.escape.url_unescape(self.get_cookie("username",u"誰かさん"))
pos = self.application.gpos(dbname,page)
- table = self.application.db[dbname]
+ table = self.application.db.table(dbname)
start = (pos-1)*i
if start < 0:
- start = table.count()-i
+ start = len(table)-i
if start < 0:
start = 0
- rec = table.find()
- rec.sort('number')
- rec.skip(start).limit(i)
- if table.count() >= 10*i:
+ rec = sorted(table.all(),key=lambda x: x['number'])[start:start+i]
+ if len(table) >= 10*i:
self.render('modules/full.htm',position=pos,records=rec,data=params,db=dbname)
return
self.render('modules/index.htm',position=pos,records=rec,data=params,username=na,db=dbname)
self.render('login.htm')
def post(self):
- pw = self.application.db['params'].find_one()
+ pw = self.application.db.get(where('kinds') == 'conf')
if self.get_argument('password') == pw['password']:
self.set_current_user('admin')
dbname = self.get_argument('record')
class NaviHandler(tornado.web.RequestHandler):
def get(self):
- coll = self.application.db.collection_names(include_system_collections=False)
- self.render('top.htm',coll=coll,full=self.full)
-
+ self.render('top.htm',coll=sorted(self.name()),full=self.full)
+
+ def name(self):
+ for x in self.application.db.tables():
+ if x != '_default':
+ yield x
+
def full(self,dbname):
- if dbname in self.application.db.collection_names(include_system_collections=False):
- i = 10*self.application.db['params'].find_one()['count']
- table = self.application.db[dbname]
- if table.count() >= i:
+ if dbname in self.application.db.tables():
+ i = 10*self.application.db.get(where('kinds') == 'conf')['count']
+ table = self.application.db.table(dbname)
+ if len(table) >= i:
return True
return False
self.render('title.htm',coll=rec,full=self.full)
def title(self):
- for x in self.application.db.collection_names(include_system_collections=False):
+ for x in self.name():
item = {}
item['name'] = x
- table = self.application.db[x]
+ table = self.application.db.table(x)
i = len(table)
item['count'] = i
- tmp = table.find_one({'number':1})
- if tmp:
- s = tmp['title']
+ if table.contains(where('number') == 1) == True:
+ s = table.get(where('number') == 1)['title']
else:
s = ''
item['title'] = s
item['date'] = ''
item['date2'] = 0
else:
- rec = table.find().sort('number')
+ rec = sorted(table.all(),key=lambda k: k['number'])
s = rec[i-1]['date']
item['date'] = s
i = datetime.strptime(s,'%Y/%m/%d %H:%M')
if self.application.collection(dbname) == False:
raise tornado.web.HTTPError(404)
return
- rec = self.application.db[dbname].find_one()
+ self.database = dbname
+ rec = self.application.db.get(where('kinds') == 'conf')
words = rec['bad_words']
out = rec['out_words']
na = self.get_argument('name')
sub = self.get_argument('title')
- com = self.get_argument('comment')
+ com = self.get_argument('comment',None,False)
text = ''
i = 0
+ url = []
error = ''
for word in out:
if word in com:
error = error + u'禁止ワード.'
break
for line in com.splitlines(True):
+ if error != '':
+ break
for word in words:
- if word in line:
+ if word in line.lower():
error = error + u'タグ違反.('+word+')'
- i += len(line)
+ i += len(line)
+ obj = re.finditer('http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', line)
+ for x in obj:
+ if x.group() not in url:
+ url.append(x.group())
+ if re.match(' ',line):
+ line = line.replace(' ',' ',1)
text = text+'<p>'+self.link(line)+'<br></p>'
+ s = ''
+ for x in url:
+ s = s+'<tr><td><a class=livepreview target=_blank href={0}>{0}</a></td></tr>'.format(x)
+ if s:
+ text = text+'<table><tr><td>検出URL:</td></tr>'+s+'</table>'
pw = self.get_argument('password')
- if sub == '':
- sub = u'タイトルなし.'
if i == 0:
error = error + u'本文がありません.'
elif i > 1000:
error = error +u'文字数が1,000をこえました.'
- article = self.application.db[dbname]
- if article.count() == 0:
+ article = self.application.db.table(dbname)
+ if len(article) == 0:
no = 1
- else:
- items = article.find()
- item = items.sort('number')[article.count()-1]
+ else:
+ item = sorted(article.all(),key=lambda x: x['number'])[len(article)-1]
no = item['number']+1
if error == '':
+ if not na:
+ na = u'誰かさん'
s = datetime.now()
reg = {'number':no,'name':na,'title':sub,'comment':text,'raw':com,'password':pw,'date':s.strftime('%Y/%m/%d %H:%M')}
article.insert(reg)
self.render('regist.htm',content=error)
def link(self,command):
- y = ''
i = 0
text = ''
- for x in command.split():
- if (y == '>>')and(x.isdecimal() == True):
- s = '<a href=#'+x+'>'+x+'</a>'
- while -1 < command.find(x,i):
- j = command.find(x,i)
- tmp = command[i:j]
- i = j+len(x)
- k = tmp.rsplit(None,1)
- if ((len(k) > 1)and(k[1] == y))or(k[0] == y):
- text = text+tmp+s
- break
- else:
- text = text+tmp+x
- y = x
- if text == '':
- return command
+ obj = re.finditer('>>[0-9]+',command)
+ for x in obj:
+ s = '<a class=minpreview data-preview-url=/{0}?key={1} href=/{0}#{1}>>>{1}</a>'.format(self.database,x.group()[2:])
+ text = text+command[i:x.start()]+s
+ i = x.end()
else:
- if len(command) > i:
- return text+command[i:]
- else:
- return text
+ text = text+command[i:]
+ return text
class AdminHandler(BaseHandler):
@tornado.web.authenticated
if self.application.collection(dbname) == False:
raise tornado.web.HTTPError(404)
return
- table = self.application.db[dbname]
- rec = table.find().sort('number')
- mente = self.application.db['params'].find_one()
+ table = self.application.db.table(dbname)
+ rec = sorted(table.all(),key=lambda x: x['number'])
+ mente = self.application.db.get(where('kinds') == 'conf')
if mente['mentenance'] == True:
check = 'checked=checked'
else:
i = mente['count']
start = (pos-1)*i
if start < 0:
- start = table.count()-i
+ start = len(table)-i
if start < 0:
start = 0
- rec.skip(start).limit(i)
- self.render('modules/admin.htm',position=pos,records=rec,mente=check,password=mente['password'],db=dbname)
+ self.render('modules/admin.htm',position=pos,records=rec[start:start+i],mente=check,password=mente['password'],db=dbname)
class AdminConfHandler(BaseHandler):
@tornado.web.authenticated
def post(self,dbname,func):
if func == 'set':
- param = self.application.db['params'].find_one()
+ param = self.application.db.get(where('kinds') == 'conf')['mentenance']
if self.get_argument('mente','') == 'on':
mente = True
+ if param != mente:
+ self.store()
else:
mente = False
+ if param != mente:
+ self.restore()
word = self.get_argument('pass','')
if word == '':
self.render('regist.htm',content='パスワードを設定してください')
return
else:
- param.update({'mentenance':mente,'password':word})
- param.save(param)
+ self.application.db.update({'mentenance':mente,'password':word},where('kinds') == 'conf')
elif func == 'del':
- table = self.application.db[dbname]
+ table = self.application.db.table(dbname)
for x in self.get_arguments('item'):
- table.remove({'number':int(x)})
+ table.remove(where('number') == int(x))
self.redirect('/'+dbname+'/admin/0/')
+
+ def store(self):
+ self.application.db.close()
+ shutil.copy(st.json,st.bak)
+ self.application.db = TinyDB(st.json)
+
+ def restore(self):
+ database = self.application.db
+ bak = TinyDB(st.bak)
+ for x in database.tables():
+ if self.application.collection(x) == True:
+ database.purge_table(x)
+ if x in bak.tables():
+ table = database.table(x)
+ table.insert_multiple(bak.table(x).all())
class UserHandler(tornado.web.RequestHandler):
def post(self,dbname):
- num = int(self.get_argument('number'))
- pas = self.get_argument('password')
- table = self.application.db[dbname]
- obj = table.find_one({'number':num})
- if obj and(obj['password'] == pas):
- table.remove({'number':num})
+ num = self.get_argument('number')
+ if num.isdigit() == True:
+ num = int(num)
+ pas = self.get_argument('password')
+ table = self.application.db.table(dbname)
+ qwr = Query()
+ obj = table.get(qwr.number == num)
+ if obj and(obj['password'] == pas):
+ table.remove(qwr.number == num)
self.redirect('/'+dbname)
class SearchHandler(tornado.web.RequestHandler):
self.render('modules/search.htm',records=[],word1=word,db=dbname)
def search(self,dbname):
- table = self.application.db[dbname]
+ table = self.application.db.table(dbname)
element = self.word.split()
if len(element) == 0:
element = ['']
while len(element) < 3:
element.append(element[0])
+ if self.radiobox == 'comment':
+ query = (Query().raw.search(element[0])) | (Query().raw.search(element[1])) | (Query().raw.search(element[2]))
+ else:
+ query = (Query().name == element[0]) | (Query().name == element[1]) | (Query().name == element[2])
if self.radiobox == 'comment':
- for x in table.find({'$or':[{'name':re.compile(element[0])},{'name':re.compile(element[1])},{'name':re.compile(element[2])}]}):
+ for x in table.search(query):
com = ''
for text in x['raw'].splitlines(True):
for word in self.word.split():
x['comment'] = com
yield x
else:
- for x in table.find({'$or':[{'name':element[0]},{'name':element[1]},{'name':element[2]}]}):
- yield x
+ for x in table.search(query):
+ yield x
class FooterModule(tornado.web.UIModule):
def render(self,number,url,link):
return self.render_string('modules/footer.htm',index=number,url=url,link=link)
+class HeadlineApi(tornado.web.RequestHandler):
+ def get(self):
+ response = {}
+ for x in self.application.db.tables():
+ if x != '_default':
+ response[x] = self.get_data(x)
+ self.write(json.dumps(response,ensure_ascii=False))
+
+ def get_data(self,dbname):
+ table = self.application.db.table(dbname)
+ i = len(table)
+ if i == 0:
+ return {}
+ else:
+ rec = sorted(table.all(),key=lambda x: x['number'])[i-1]
+ return {'number':rec['number'],'title':rec['title'],'name':rec['name'],'comment':rec['raw'][0:19]}
+
+class ArticleApi(tornado.web.RequestHandler):
+ def get(self,dbname,number):
+ if self.application.collection(dbname) == True:
+ table = self.application.db.table(dbname)
+ response = table.get(where('number') == int(number))
+ if response == None:
+ response = {}
+ else:
+ del response['comment']
+ self.write(json.dumps(response,ensure_ascii=False))
+ else:
+ tornado.web.HTTPError(404)
+
+ def post(self,dbname):
+ name = self.get_argument('name',u'誰かさん')
+ title = self.get_argument('title',u'タイトルなし')
+ comment = self.get_argument('comment')
+ table = self.application.db.table(dbname)
+ table.insert({'name':name,'title':title,'comment':comment})
+
class Application(tornado.web.Application):
def __init__(self):
+ self.db = TinyDB(st.json)
handlers = [(r'/',NaviHandler),(r'/login',LoginHandler),(r'/logout',LogoutHandler),(r'/title',TitleHandler),
+ (r'/headline/api',HeadlineApi),(r'/read/api/([a-zA-Z0-9_]+)/([0-9]+)',ArticleApi),(r'/write/api/([a-zA-Z0-9_]+)',ArticleApi),
(r'/([a-zA-Z0-9_]+)',IndexHandler),(r'/([a-zA-Z0-9_]+)/([0-9]+)/',IndexHandler),
(r'/([a-zA-Z0-9_]+)/admin/([0-9]+)/',AdminHandler),(r'/([a-zA-Z0-9_]+)/admin/([a-z]+)/',AdminConfHandler),(r'/([a-zA-Z0-9_]+)/userdel',UserHandler),
(r'/([a-zA-Z0-9_]+)/search',SearchHandler),(r'/([a-zA-Z0-9_]+)/regist',RegistHandler)]
'ui_modules':{'Footer':FooterModule},
'cookie_secret':'bZJc2sWbQLKos6GkHn/VB9oXwQt8SOROkRvJ5/xJ89E=',
'xsrf_cookies':True,
- #'debug':True,
+ 'debug':True,
'login_url':'/login'
}
tornado.web.Application.__init__(self,handlers,**settings)
def gpos(self,dbname,page):
- params = self.db['params'].find_one()
+ params = self.db.get(where('kinds') == 'conf')
pos = int(page)
if pos <= 0:
pos = 0
- elif (pos-1)*params['count'] >= self.db[dbname].count():
+ elif (pos-1)*params['count'] >= len(self.db.table(dbname)):
pos = 0
return pos
def collection(self,name):
- if name in self.db.collection_names():
+ if name in self.db.tables():
return True
else:
return False
-app = Application()
-MONGOLAB_URI = os.environ.get('mongodb://kainushi:1234abcd@ds113678.mlab.com:13678/heroku_n905jfw2')
-if MONGOLAB_URI:
- conn = pymongo.MongoClient(MONGOLAB_URI)
- app.db = conn.heroku_n905jfw2
-else:
- conn = pymongo.MongoClient()
- app.db = conn.mydatabase
-
+class static():
+ json = 'static/db/db.json'
+ bak = 'static/db/bak.json'
+
+st = static()
+if __name__ == '__main__':
+ tornado.options.parse_command_line()
+ http_server = tornado.httpserver.HTTPServer(Application())
+ http_server.listen(options.port)
+ tornado.ioloop.IOLoop.instance().start()
OSDN Git Service
