pybbs.py

   1 # -*- coding:utf-8 -*-
   2 import os,re,glob
   3 from tornado import escape,web,ioloop,httpserver,httpclient
   4 import pymongo, urllib
   5 from datetime import datetime
   6 import json
   7 from bson.objectid import ObjectId #don't remove
   8 from linebot.api import LineBotApi
   9 from linebot.exceptions import (InvalidSignatureError)
  10 from linebot.models import (TextSendMessage)
  11
  12
  13 class BaseHandler(web.RequestHandler):
  14     def get_current_user(self):
  15         user = self.get_secure_cookie('admin_user')
  16         return escape.utf8(user)
  17
  18     def set_current_user(self,username):
  19         self.set_secure_cookie('admin_user',username)
  20
  21     def clear_current_user(self):
  22         self.clear_cookie('admin_user')
  23
  24 class IndexHandler(BaseHandler):
  25     def main(self,dbname,page):
  26         params = self.application.db['params'].find_one({'app':'bbs'})
  27         if params['mentenance'] is True:
  28             self.render('mentenance.htm',title=params['title'],db=dbname)
  29             return
  30         if dbname not in self.application.mylist():
  31             if self.current_user == b'admin':
  32                 coll = self.application.db[dbname]
  33                 coll.insert({})
  34                 coll.remove({})
  35             else:
  36                 raise web.HTTPError(404)
  37         key = self.get_argument('key','')
  38         if key != '':
  39             table = self.application.db[dbname]
  40             rec = table.find_one({'number':int(key)})
  41             if rec:
  42                 self.render('article.htm',record=rec)
  43                 return
  44             else:
  45                 raise web.HTTPError(404)
  46         self.rule = escape.url_unescape(self.get_cookie('aikotoba',''))
  47         self.na = escape.url_unescape(self.get_cookie('username',u'誰かさん'))
  48         self.pos = self.application.gpos(dbname,page)
  49         table = self.application.db[dbname]
  50         i = params['count']
  51         start = (self.pos-1)*i
  52         if start < 0:
  53             start = table.count()-i
  54             if start < 0:
  55                 start = 0
  56         rec = table.find()
  57         self.bool = (dbname == params['info name'])
  58         if self.bool is True:
  59             rec.sort('number',-1)
  60         else:
  61             rec.sort('number')
  62         self.rec = rec.skip(start).limit(i)
  63
  64     def get(self,dbname,page='0'):
  65         self.main(dbname,page)
  66         db = self.application.db
  67         table = db[dbname].find()
  68         params = db['params'].find_one({'app':'bbs'})
  69         if table.count() >= 10*params['count']:
  70             self.render('modules/full.htm',position=self.pos,records=self.rec,data=params,db=dbname)
  71         if self.bool is True and self.current_user != b'admin':
  72             self.render('modules/info.htm',position=self.pos,records=self.rec,data=params,db=dbname,error='')
  73         else:
  74             self.render('modules/index.htm',position=self.pos,records=self.rec,data=params,username=self.na,
  75             comment='',db=dbname,aikotoba=self.rule,error='',check='checked')
  76
  77 class LoginHandler(BaseHandler):
  78     def get(self):
  79         info = self.application.db['params'].find_one({'app':'bbs'})
  80         query = self.get_query_argument('next','/'+info['info name'])
  81         i = query[1:].find('/')
  82         if i == -1:
  83             qs = query[1:]
  84         else:
  85             qs = query[1:i+1]
  86         self.render('login.htm',db=escape.url_unescape(qs))
  87
  88     def post(self):
  89         dbname = self.get_argument('record','')
  90         if dbname == '':
  91             self.redirect('/login')
  92             return
  93         pw = self.application.db['params'].find_one({'app':'bbs'})
  94         if self.get_argument('password') == pw['password']:
  95             self.set_current_user('admin')
  96         if dbname == 'master':
  97             self.redirect('/master')
  98         else:
  99             self.redirect('/'+dbname+'/admin/0/')
 100
 101 class LogoutHandler(BaseHandler):
 102     def get(self):
 103         self.clear_current_user()
 104         self.redirect('/login')
 105
 106 class JumpHandler(BaseHandler):
 107     def get(self):
 108         self.clear_current_user()
 109         self.redirect('/')
 110
 111 class NaviHandler(web.RequestHandler):
 112     def get(self):
 113         if 'params' not in self.application.mylist():
 114             item = {"mentenance":False,"out_words":[u"阿保",u"馬鹿",u"死ね"],"password":"admin",
 115                     "title2":"<h1 style=color:gray;text-align:center>pybbs</h1>",
 116                     "bad_words":["<style","<link","<script","<img"],"count":30,
 117                     "title":"pybbs","info name":"info",'app':'bbs'}
 118             self.application.db['params'].insert(item)
 119             self.application.db['info'].find()
 120         table = self.application.db['params'].find_one({'app':'bbs'})
 121         if table['mentenance'] is True:
 122             self.render('mentenance.htm',title=table['title'],db=table['info name'])
 123             return
 124         coll = self.application.coll()
 125         na = table['info name']
 126         self.render('top.htm',coll=coll,name=na,full=self.full)
 127
 128     def full(self,dbname):
 129         if dbname in self.application.coll():
 130             i = 10*self.application.db['params'].find_one({'app':'bbs'})['count']
 131             table = self.application.db[dbname]
 132             if table.count() >= i:
 133                 return True
 134         return False
 135
 136 class TitleHandler(NaviHandler):
 137     def get(self):
 138         rec = sorted(self.title(),key=lambda x: x['date2'])
 139         self.render('title.htm',coll=rec,full=self.full)
 140
 141     def title(self):
 142         for x in self.application.coll():
 143             item = {}
 144             item['name'] = x
 145             table = self.application.db[x]
 146             i = table.count()
 147             item['count'] = i
 148             tmp = table.find_one({'number':1})
 149             if tmp:
 150                 s = tmp['title']
 151             else:
 152                 s = ''
 153             item['title'] = s
 154             if i == 0:
 155                 item['date'] = ''
 156                 item['date2'] = 0
 157             else:
 158                 rec = table.find().sort('number')
 159                 s = rec[i-1]['date']
 160                 item['date'] = s
 161                 i = datetime.strptime(s,'%Y/%m/%d %H:%M')
 162                 year = datetime.now().year-i.year
 163                 if year == 0:
 164                     j = 800
 165                 elif year == 1:
 166                     j = 400
 167                 else:
 168                     j = 0
 169                 item['date2'] = j+31*(i.month-1)+i.day
 170             yield item
 171
 172 class RegistHandler(IndexHandler):
 173     def post(self,dbname):
 174         self.main(dbname,'0')
 175         params = self.application.db['params'].find_one({'app':'bbs'})
 176         if dbname not in self.application.coll() and dbname != params['info name']:
 177             raise web.HTTPError(404)
 178         words = params['bad_words']
 179         out = params['out_words']
 180         rule = self.get_argument('aikotoba')
 181         na = self.get_argument('name')
 182         sub = self.get_argument('title')
 183         com = self.get_argument('comment',None,False)
 184         text = ''
 185         i = 0
 186         url = []
 187         error = ''
 188         kinsi = False
 189         for line in com.splitlines():
 190             if kinsi is False:
 191                 for word in out:
 192                     if word in line:
 193                         error += u'禁止ワード.<br>'
 194                         kinsi = True
 195                         break
 196             for word in words:
 197                 if word in line.lower():
 198                     tag = escape.xhtml_escape(word)
 199                     error += u'タグ違反.('+tag+')<br>'
 200             i += len(line)
 201             obj = re.finditer('http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+', line)
 202             for x in obj:
 203                 if x.group() not in url:
 204                     url.append(x.group())
 205             j = 0
 206             for x in line:
 207                 if x == ' ':
 208                     j += 1
 209                 else:
 210                     break
 211             if j > 0:
 212                 line = line.replace(' ','&nbsp;',j)
 213             if len(line) == 0:
 214                 text += '<p><br>\n</p>'
 215             else:
 216                 text += '<p>'+self.link(line,dbname)+'\n</p>'
 217         if rule != u'げんき':
 218             error += u'合言葉未入力.<br>'
 219         s = ''
 220         for x in url:
 221             s = s+'<tr><td><a href={0} class=livepreview target=_blank>{0}</a></td></tr>'.format(x)
 222         if s != '':
 223             text = text+'<table><tr><td>検出url:</td></tr>'+s+'</table>'
 224         pw = self.get_argument('password')
 225         if i > 1000:
 226             error += u'文字数が1,000をこえました.<br>'
 227         if na == '':
 228             na = u'誰かさん'
 229         if sub == '':
 230             sub = u'タイトルなし.'
 231         article = self.application.db[dbname]
 232         if article.count() == 0:
 233             no = 1
 234         else:
 235             items = article.find()
 236             item = items.sort('number')[article.count()-1]
 237             no = item['number']+1
 238         if error == '':
 239             if self.get_argument('show','false') == 'true':
 240                 error = '<p style=font-size:2.5em;collor:blue>↓↓プレビュー↓↓<br></p>'+text
 241         else:
 242             error = '<p style=color:red>'+error+'</p>'
 243         if error == '':
 244             self.set_cookie('aikotoba',escape.url_escape(rule))
 245             s = datetime.now()
 246             reg = {'number':no,'name':na,'title':sub,'comment':text,'raw':com,'password':pw,'date':s.strftime('%Y/%m/%d %H:%M')}
 247             article.insert(reg)
 248             self.set_cookie('username',escape.url_escape(na))
 249             self.redirect('/'+dbname+'#article')
 250         else:
 251             error = '<p style=color:red>'+error+'</p>'
 252             self.render('modules/index.htm',position=0,records=self.rec,data=params,
 253                 username=na,comment=com,db=dbname,aikotoba=rule,error=error,check='')
 254
 255     def link(self,command,database):
 256         i = 0
 257         text = ''
 258         obj = re.finditer('>>[0-9]+',command)
 259         for x in obj:
 260             s = '<a class=minpreview data-preview-url=/{0}?key={1} href=/{0}/userdel?job={1}>>>{1}</a>'.format(database,x.group()[2:])
 261             text = text+command[i:x.start()]+s
 262             i = x.end()
 263         else:
 264             text = text+command[i:]
 265         return text
 266
 267 class AdminHandler(BaseHandler):
 268     @web.authenticated
 269     def get(self,dbname,page='0'):
 270         if dbname == '':
 271             dbname = self.get_argument('record','')
 272         table = self.application.db[dbname]
 273         rec = table.find().sort('number')
 274         mente = self.application.db['params'].find_one({'app':'bbs'})
 275         if mente['mentenance'] is True:
 276             check = 'checked=checked'
 277         else:
 278             check = ''
 279         pos = self.application.gpos(dbname,page)
 280         i = mente['count']
 281         start = (pos-1)*i
 282         if start < 0:
 283             start = table.count()-i
 284             if start < 0:
 285                 start = 0
 286         rec.skip(start).limit(i)
 287         self.render('modules/admin.htm',position=pos,records=rec,mente=check,password=mente['password'],db=dbname)
 288
 289 class AdminConfHandler(BaseHandler):
 290     @web.authenticated
 291     def post(self,dbname,func):
 292         if func == 'set':
 293             param = self.application.db['params'].find_one({'app':'bbs'})
 294             if self.get_argument('mente','') == 'on':
 295                 mente = True
 296             else:
 297                 mente = False
 298             word = self.get_argument('pass','')
 299             if word == '':
 300                 self.render('regist.htm',content='パスワードを設定してください')
 301                 return
 302             else:
 303                 param['mentenance']=mente
 304                 param['password']=word
 305                 self.application.db['params'].save(param)
 306         elif func == 'del':
 307             table = self.application.db[dbname]
 308             for x in self.get_arguments('item'):
 309                 table.remove({'number':int(x)})
 310         self.redirect('/'+dbname+'/admin/0/')
 311
 312 class UserHandler(web.RequestHandler):
 313     def get(self,dbname):
 314         q = self.get_query_argument('job','0',strip=True)
 315         self.redirect(self.application.page(dbname,q))
 316
 317     def post(self,dbname):
 318         number = self.get_argument('number')
 319         if number.isdigit() is True:
 320             num = int(number)
 321             url = self.application.page(dbname,number)
 322             if 'password' in self.request.arguments.keys():
 323                 pas = self.get_argument('password')
 324             else:
 325                 self.redirect(url)
 326                 return
 327             table = self.application.db[dbname]
 328             obj = table.find_one({'number':num})
 329             if obj and(obj['password'] == pas):
 330                 table.update({'number':num},{'$set':{'title':u'削除されました','name':'','comment':u'<i><b>投稿者により削除されました</b></i>','raw':''}})
 331                 self.redirect(url)
 332                 return
 333         self.redirect('/'+dbname)
 334
 335 class SearchHandler(web.RequestHandler):
 336     def post(self,dbname=''):
 337         arg = self.get_argument('word1')
 338         self.word = arg[:]
 339         self.andor = self.get_argument('type')
 340         self.radiobox = self.get_argument('filter')
 341         if dbname == '':
 342             rec = []
 343             for x in self.application.coll():
 344                 moji = self.search(x)
 345                 for y in sorted(moji,key=lambda k: k['number']):
 346                     y['dbname'] = x
 347                     rec.append(y)
 348         else:
 349             rec = sorted(self.search(dbname),key=lambda x: x['number'])
 350         self.render('modules/search.htm',records=rec,word1=arg,db=dbname)
 351
 352     def get(self,dbname=''):
 353         if dbname not in self.application.coll() and dbname != '':
 354             raise web.HTTPError(404)
 355         self.render('modules/search.htm',records=[],word1='',db=dbname)
 356
 357     def search(self,dbname):
 358         table = self.application.db[dbname]
 359         andor = self.andor == 'OR'
 360         element = self.word.split()
 361         elm = []
 362         for x in element:
 363             if x != '':
 364                 elm.append(re.escape(x))
 365         if self.radiobox == 'comment':
 366             query = []
 367             for qu in elm:
 368                 query.append({'raw':re.compile(qu,re.IGNORECASE)})
 369             if len(query) == 0:
 370                 return
 371             if andor:
 372                 result = table.find({'$or':query})
 373                 color = 'yellow'
 374             else:
 375                 result = table.find({'$and':query})
 376                 color = 'aqua'
 377             for x in result:
 378                 com = ''
 379                 i = 0
 380                 for text in x['raw'].splitlines():
 381                     for y in text:
 382                         if y == ' ':
 383                             i += 1
 384                         else:
 385                             break
 386                     text = text.replace(' ','&nbsp;',i)
 387                     for y in element:
 388                         if y.lower() in text.lower():
 389                             com += '<p style=background-color:'+color+'>'+text+'<br></p>'
 390                             break
 391                     else:
 392                         if text == '':
 393                             com += '<br>'
 394                         else:
 395                             com += '<p>'+text+'</p>'
 396                 x['comment'] = com
 397                 yield x
 398         else:
 399             query = []
 400             for x in element:
 401                 if x != '':
 402                     query.append({'name':x})
 403             if len(query) == 0:
 404                 return
 405             for x in table.find({'$or':query}):
 406                 yield x
 407
 408 class HelpHandler(web.RequestHandler):
 409     def get(self):
 410         self.render('help.htm',req='')
 411
 412     def post(self):
 413         com = self.get_argument('help','')
 414         line = com.splitlines(True)
 415         com = ''
 416         for x in line:
 417             com += '<p>'+x
 418         time = datetime.now()
 419         db = self.application.db['master']
 420         db.insert({'comment':com,'time':time.strftime('%Y/%m/%d')})
 421         self.render('help.htm',req='送信しました')
 422
 423 class MasterHandler(BaseHandler):
 424     @web.authenticated
 425     def get(self):
 426         if self.current_user == b'admin':
 427             com = self.application.db['master'].find()
 428             sum = self.application.db['temp'].find().count()
 429             self.render('master.htm',com=com,sum=sum)
 430         else:
 431             raise web.HTTPError(404)
 432
 433 class AlertHandler(web.RequestHandler):
 434     def get(self):
 435         db = self.get_query_argument('db')
 436         num = self.get_query_argument('num')
 437         table = self.application.db[db]
 438         tb = table.find_one({'number':int(num)})
 439         com = tb['comment']
 440         time = datetime.now().strftime('%Y/%m/%d')
 441         link = self.application.page(db,num)
 442         jump = '<p><a href={0}>{0}</a>'.format(link)
 443         d = datetime.now().weekday()
 444         table = self.application.db['temp']
 445         table.remove({'date':{'$ne':d}})
 446         result = table.insert(
 447             {'comment':com+jump,'time':time,'link':link,'date':d,'db':db,'num':num})
 448         self.render('alert.htm',com=com+jump,num=str(result))
 449
 450     def post(self):
 451         id = ObjectId(self.get_argument('num'))
 452         table = self.application.db['temp']
 453         tb = table.find_one({'_id':id})
 454         link = tb['link']
 455         com = self.get_argument('com')
 456         table.remove({'_id':id})
 457         if self.get_argument('cancel','') == 'cancel':
 458             self.redirect(link)
 459             return
 460         if com != '':
 461             tb['comment'] = com+tb['comment']
 462         del tb['date']
 463         table = self.application.db['master']
 464         table.insert(tb)
 465         self.redirect(link)
 466
 467 class CleanHandler(web.RequestHandler):
 468     def post(self):
 469         bool = self.get_argument('all', 'false').lower()
 470         table = self.application.db['master']
 471         if bool == 'true':
 472             table.remove()
 473             self.application.db['temp'].remove()
 474         elif bool == 'false':
 475             for x in list(table.find()):
 476                 if not 'num' in x.keys():
 477                     table.remove({'_id':x['_id']})
 478                 else:
 479                     item = self.application.db[x['db']].find_one({'number':int(x['num'])})
 480                     if (not item)or(item['raw'] == ''):
 481                         table.remove({'_id':x['_id']})
 482         com = self.application.db['master'].find()
 483         sum = self.application.db['temp'].find().count()
 484         self.render('master.htm', com=com, sum=sum)
 485
 486 class FooterModule(web.UIModule):
 487     def render(self,number,url,link):
 488         return self.render_string('modules/footer.htm',index=number,url=url,link=link)
 489
 490 class HeadlineApi(web.RequestHandler):
 491     def get(self):
 492         response = {}
 493         for coll in self.application.coll():
 494             table = self.application.db[coll]
 495             if table.count() == 0:
 496                 mydict = {}
 497             else:
 498                 text = table.find().sort('number')[table.count()-1]
 499                 mydict = {'number':text['number'],'name':text['name'],'title':text['title'],'comment':text['raw'][0:20]}
 500             response[coll] = mydict
 501         self.write(json.dumps(response,ensure_ascii=False))
 502
 503 class ArticleApi(web.RequestHandler):
 504     def get(self,dbname,number):
 505         response = None
 506         if dbname in self.application.coll():
 507             table = self.application.db[dbname]
 508             response = table.find_one({'number':int(number)})
 509         if not response:
 510             response = {}
 511         else:
 512             del response['_id']
 513             del response['comment']
 514             del response['password']
 515         self.write(json.dumps(response,ensure_ascii=False))
 516
 517 class ListApi(web.RequestHandler):
 518     def get(self,dbname):
 519         response = None
 520         if dbname in self.application.coll():
 521             table = self.application.db[dbname]
 522             response = {}
 523             for data in table.find().sort('number'):
 524                 response[data['number']] = data['raw'][0:20]
 525         if not response:
 526             response = {}
 527         self.write(json.dumps(response,ensure_ascii=False))
 528
 529 class WebHookHandler(web.RequestHandler):
 530     def main(self, no):
 531         #pz = pytz.timezone('Asia/Tokyo')
 532         now = datetime.now()#pz)
 533         t = now.hour
 534         w = now.weekday()
 535         if (w < 5)and(t >= 9)and(t < 16):
 536             return u'仕事中.'
 537         table, na = self.users()
 538         item = table.find({'no':re.compile(no,re.IGNORECASE)})
 539         if item.count() == 1:
 540             x = item[0]
 541             ans = x['name']+'\n'+x['no']
 542         elif item.count() > 1:
 543             ans = ''
 544             obj = list(item)
 545             list1 = sorted(obj, key=lambda k:k['name'])
 546             for x in list1:
 547                 if x['name'] == list1[0]['name']:
 548                     ans += x['name']+'\n'+x['no']+'\n'
 549                 else:
 550                     break
 551             else:
 552                 return ans
 553             ans = self.itr(sorted(list1, key=lambda k:k['no']))
 554         else:
 555             ans = self.itr(table.find().sort('no'))
 556             ans = '-*-'+na+' list-*-\n'+ans
 557         return ans
 558
 559     def itr(self, item):
 560         ans = ''
 561         for x in item:
 562             ans += '【'+x['no']+'】 '
 563         return ans
 564
 565     def help(self):
 566         s = '-*-database names-*-\n'
 567         out = ['objectlabs-system','objectlabs-system.admin.collections','users_bot']
 568         for x in self.application.mylist():
 569             if x not in out and x[-4:] == '_bot' and x != '_bot':
 570                 s += x[:-4]+'\n'
 571         return s
 572
 573     def setting(self, dbname):
 574         dbname = dbname.lower()+'_bot'
 575         ca = self.application.mylist()
 576         if 'users_bos' in ca:
 577             ca.remove('users_bot')
 578         if dbname in ca:
 579             db = self.application.db['users_bot']
 580             item = db.find_one({'name':self.uid})
 581             if item['dbname'] == dbname:
 582                 return False
 583             else:
 584                 db.update({'name':self.uid}, {'name':self.uid, 'dbname':dbname})
 585                 return True
 586         return False
 587
 588     def users(self):
 589         db = self.application.db['users_bot']
 590         item = db.find_one({'name':self.uid})
 591         x = item['dbname']
 592         return self.application.db[x], x[:-4]
 593
 594     def post(self):
 595         '''
 596         signature = self.request.headers['X-Line-Signature']
 597         body = self.request.body
 598         parser = WebhookParser(self.application.ch)
 599         try:
 600             parser.parse(body, signature)
 601         except InvalidSignatureError:
 602             web.HTTPError(404)
 603             return
 604         '''
 605         dic = escape.json_decode(self.request.body)
 606         for event in dic['events']:
 607             if 'replyToken' in event.keys():
 608                 self.uid = event['source']['userId']
 609                 bot = 'users_bot'
 610                 if event['type'] == 'unfollow':
 611                     self.application.db[bot].remove({'name':self.uid})
 612                     return
 613                 elif event['type'] != 'message' or event['message']['type'] != 'text':
 614                     return
 615                 item = self.application.db['params'].find_one({'app':'bot'})
 616                 if item:
 617                     de = item['default']
 618                 else:
 619                     de = '_bot'
 620                 if item and 'access_token' in item.keys():
 621                     token = item['access_token']
 622                 else:
 623                     token =self.application.tk
 624                 if bot not in self.application.mylist() or not self.application.db[bot].find_one({'name':self.uid}):
 625                     db = self.application.db[bot]
 626                     db.insert({'name':self.uid, 'dbname':de})
 627                 x = event['message']['text']
 628                 if self.setting(x):
 629                     te = u'設定完了.'
 630                 elif x == '?':
 631                     te = self.help()
 632                 else:
 633                     te = self.main(x)
 634                 linebot = LineBotApi(token)
 635                 linebot.reply_message(event['replyToken'], TextSendMessage(text=te))
 636
 637 class InitHandler(web.RequestHandler):
 638     def get(self):
 639         de = self.get_argument('default', '')
 640         if de == '':
 641             names = self.application.mylist()
 642             db = []
 643             for x in names:
 644                 if x[-4:] == '_bot' and x != 'users_bot':
 645                     db.append(x[:-4])
 646             self.render('init.htm',db=db)
 647             return
 648         tb = self.application.db['params']
 649         if tb.find_one({'app':'bot'}):
 650             tb.update({'app':'bot'}, {'app':'bot', 'default':de+'_bot'})
 651         else:
 652             tb.insert({'app':'bot', 'default':de+'_bot'})
 653         for x in glob.glob('./*.txt'):
 654             f = open(x)
 655             data = f.read()
 656             f.close()
 657             self.main(x[2:-4].lower(), data)
 658
 659     def main(self, name, data):
 660         if name == 'requirements':
 661             return
 662         item = []
 663         dic = None
 664         for x in data.split('\n'):
 665             if len(x) > 0 and x[0] == '@':
 666                 dic = {}
 667                 dic['name'] = x[1:]
 668             elif dic:
 669                 dic['no'] = x
 670                 item.append(dic)
 671         table = self.application.db[name+'_bot']
 672         table.remove()
 673         for x in item:
 674             table.insert(x)
 675
 676 class TokenHandler(web.RequestHandler):
 677     def on_response(self, response):
 678         dic = escape.json_decode(response.body)
 679         token = dic['access_token']
 680         table = self.application.db['params']
 681         data = {'app':'bot', 'access_token':token}
 682         if table.find_one({'app':'bot'}):
 683             table.save(data)
 684         else:
 685             table.insert(data)
 686         self.finish()
 687
 688     #@web.asynchronous
 689     def get(self):
 690         url = 'https://api.line.me/v2/oauth/accessToken'
 691         headers = 'application/x-www-form-urlencoded'
 692         data = {'grant_type':'client_credentials', 'client_id':self.application.id, 'client_secret':self.application.ch}
 693         body = urllib.parse.urlencode(data)
 694         req = httpclient.HTTPRequest(url=url,method='POST',headers=headers,body=body)
 695         http = httpclient.AsyncHTTPClient()
 696         http.fetch(req, callback=self.on_response)
 697
 698 class Application(web.Application):
 699     ch = os.environ['Channel_Secret']
 700     uri = os.environ['MONGODB_URI']
 701     ac = os.environ['ACCOUNT']
 702     tk = os.environ['Access_Token']
 703     db = pymongo.MongoClient(uri)[ac]
 704     def __init__(self):
 705         handlers = [(r'/',NaviHandler),(r'/login',LoginHandler),(r'/logout',LogoutHandler),(r'/title',TitleHandler),
 706                     (r'/headline/api',HeadlineApi),(r'/read/api/([a-zA-Z0-9_%]+)/([0-9]+)',ArticleApi),
 707                     (r'/write/api/([a-zA-Z0-9_%]+)/()/()/()',ArticleApi),(r'/list/api/([a-zA-Z0-9_%]+)',ListApi),
 708                     (r'/help',HelpHandler),(r'/master',MasterHandler),(r'/alert',AlertHandler),(r'/jump',JumpHandler),
 709                     (r'/callback',WebHookHandler),(r'/init',InitHandler),(r'/search',SearchHandler),(r'/clean',CleanHandler),(r'/token',TokenHandler),
 710                     (r'/([a-zA-Z0-9_%]+)',IndexHandler),(r'/([a-zA-Z0-9_%]+)/([0-9]+)/',IndexHandler),
 711                     (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),
 712                     (r'/([a-zA-Z0-9_%]+)/search',SearchHandler),(r'/([a-zA-Z0-9_%]+)/regist',RegistHandler)]
 713         settings = {'template_path':os.path.join(os.path.dirname(__file__),'templates'),
 714                         'static_path':os.path.join(os.path.dirname(__file__),'static'),
 715                         'ui_modules':{'Footer':FooterModule},
 716                         'cookie_secret':os.environ['cookie'],
 717                         'xsrf_cookies':False,
 718                         #'debug':True,
 719                         'login_url':'/login'
 720                         }
 721         super().__init__(handlers,**settings)
 722
 723     def gpos(self,dbname,page):
 724         params = self.db['params'].find_one({'app':'bbs'})
 725         pos = int(page)
 726         if pos <= 0:
 727             pos = 0
 728         elif (pos-1)*params['count'] >= self.db[dbname].count():
 729             pos = 0
 730         return pos
 731
 732     def page(self,dbname,number):
 733         table = self.db[dbname]
 734         rec = table.find({'number':{'$lte':int(number)}}).count()
 735         s = self.db['params'].find_one({'app':'bbs'})
 736         conf = int(s['count'])
 737         if table.find().count() - rec >= conf:
 738             return '/'+dbname+'/'+str(1+rec//conf)+'/#'+number
 739         else:
 740             return '/'+dbname+'#'+number
 741
 742     def mylist(self):
 743         return self.db.list_collection_names()[:]
 744
 745     def coll(self):
 746         name = self.mylist()
 747         item = self.db['params'].find_one({'app':'bbs'})
 748         target = ['objectlabs-system', 'objectlabs-system.admin.collections', 'system.indexes',
 749             'params', 'master', 'temp', item['info name']]
 750         for x in target:
 751             name.remove(x)
 752         for x in name:
 753             if x[-4:] == '_bot':
 754                 name.remove(x)
 755         return sorted(name)
 756
 757 if __name__ == '__main__':
 758     app = Application()
 759     http_server = httpserver.HTTPServer(app)
 760     port = int(os.environ.get('PORT',5000))
 761     http_server.listen(port)
 762     ioloop.IOLoop.instance().start()