def do_wiki(): print('==============================') print('For Wiki ... ') from torcms.model.wiki_model import MWiki mwiki = MWiki() # from torcms.model.page_model import MPage mpage = MPage() from model_ent.wiki_model import MWiki as OldWiki from model_ent.page_model import MPage as OldPage oldwiki = OldWiki() oldpage = OldPage() print('got it') abc = oldwiki.query_all(limit_num=2000) ded = oldpage.query_all() for rec in oldwiki.query_all(limit_num=20000): post_data = { "uid": '_' + ''.join(rec.uid.split('-')), "title": rec.title, "date": rec.date, "time_create": rec.time_create, "user_name": rec.user_name, "time_update": rec.time_update, "view_count": rec.view_count, "cnt_md": unescape(rec.cnt_md), "cnt_html": rec.cnt_html, "kind": '1', } # print(rec.title) mwiki.insert_data(post_data) for rec in ded: print(rec.slug) post_data = { "uid": rec.slug, "title": rec.title, "date": rec.date, "time_create": rec.time_create, "user_name": '', "time_update": rec.time_update, "view_count": rec.view_count, "cnt_md": unescape(rec.cnt_md), "cnt_html": rec.cnt_html, "kind": '2', } # print(rec.title) mwiki.insert_data(post_data) print('QED')
class WikiHandler(BaseHandler): def initialize(self): self.init() self.mwiki = MWiki() self.mwiki_hist = MWikiHist() def get(self, url_str=''): url_arr = self.parse_url(url_str) if url_str == 'recent': self.recent() elif url_arr[0] == 'ajax_count_plus': self.ajax_count_plus(url_arr[1]) elif url_str == 'refresh': self.refresh() elif url_arr[0] == 'edit': self.to_modify(url_arr[1]) elif len(url_arr) == 1: self.wiki(url_str) else: kwd = { 'info': '页面未找到', } self.render('html/404.html', kwd=kwd) def post(self, url_str=''): url_arr = self.parse_url(url_str) if url_arr[0] == 'edit': self.update(url_arr[1]) elif url_arr[0] == 'add': self.wikinsert() else: self.redirect('html/404.html') def recent(self): kwd = { 'pager': '', 'unescape': tornado.escape.xhtml_unescape, 'title': '最近文档', } self.render( 'doc/wiki/wiki_list.html', view=self.mwiki.query_recent(), format_date=tools.format_date, cfg=config.cfg, kwd=kwd, userinfo=self.userinfo, ) def refresh(self): kwd = { 'pager': '', 'unescape': tornado.escape.xhtml_unescape, 'title': '最近文档', } self.render( 'doc/wiki/wiki_list.html', view=self.mwiki.query_dated(16), format_date=tools.format_date, kwd=kwd, cfg=config.cfg, userinfo=self.userinfo, ) def wiki(self, title): dbdate = self.mwiki.get_by_wiki(title) if dbdate: self.viewit(dbdate) else: self.to_add(title) @tornado.web.authenticated def to_add(self, title): if self.check_doc_priv(self.userinfo)['ADD']: pass else: return False kwd = { 'title': title, 'pager': '', } self.render( 'doc/wiki/wiki_add.html', kwd=kwd, cfg=config.cfg, userinfo=self.userinfo, ) @tornado.web.authenticated def update(self, uid): raw_data = self.mwiki.get_by_id(uid) if self.check_doc_priv( self.userinfo )['EDIT'] or raw_data.user_name == self.get_current_user(): pass else: return False post_data = {} for key in self.request.arguments: post_data[key] = self.get_arguments(key) post_data['user_name'] = self.get_current_user() self.mwiki.update(uid, post_data) self.mwiki_hist.insert_data(raw_data) self.redirect('/wiki/{0}'.format( tornado.escape.url_escape(post_data['title'][0]))) @tornado.web.authenticated def to_modify(self, id_rec): wiki_rec = self.mwiki.get_by_id(id_rec) # 用户具有管理权限,或文章是用户自己发布的。 if self.check_doc_priv( self.userinfo )['EDIT'] or wiki_rec.user_name == self.get_current_user(): pass else: return False kwd = { 'pager': '', } self.render( 'doc/wiki/wiki_edit.html', kwd=kwd, unescape=tornado.escape.xhtml_unescape, dbrec=wiki_rec, cfg=config.cfg, userinfo=self.userinfo, ) def viewit(self, view): kwd = { 'pager': '', 'editable': self.editable(), } self.render( 'doc/wiki/wiki_view.html', view=view, unescape=tornado.escape.xhtml_unescape, kwd=kwd, userinfo=self.userinfo, cfg=config.cfg, ) def ajax_count_plus(self, slug): output = { 'status': 1 if self.mwiki.update_view_count(slug) else 0, } return json.dump(output, self) @tornado.web.authenticated def wikinsert(self): if self.check_doc_priv(self.userinfo)['ADD']: pass else: return False post_data = {} for key in self.request.arguments: post_data[key] = self.get_arguments(key) post_data['user_name'] = self.get_current_user() if self.mwiki.get_by_wiki(post_data['title'][0]): pass else: self.mwiki.insert_data(post_data) self.redirect('/wiki/{0}'.format( tornado.escape.url_escape(post_data['title'][0])))
class TestWiki(): def setup(self): print('setup 方法执行于本类中每条用例之前') self.uu = MWiki() self.raw_count = self.uu.get_counts() self.wiki_title = 'lkablkjcdefg' def test_insert(self): raw_count = self.uu.get_counts() post_data = { 'title': self.wiki_title, 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', } if self.uu.insert_data(post_data): new_count = self.uu.get_counts() tt = self.uu.get_by_wiki(self.wiki_title) assert tt.title == post_data['title'] assert tt.cnt_md == tornado.escape.xhtml_unescape( post_data['cnt_md']) assert raw_count + 1 == new_count def test_insert_2(self): '''Wiki insert: Test invalid title''' post_data = { 'title': '', 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', } uu = self.uu.insert_data(post_data) assert uu == False post_data = { 'title': '1', 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', } uu = self.uu.insert_data(post_data) assert uu == False post_data = { 'title': '天', 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', } uu = self.uu.insert_data(post_data) assert uu == False def test_get_by_title(self): post_data = { 'title': self.wiki_title, 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', } uid = self.uu.insert_data(post_data) ss = self.uu.get_by_uid(uid) assert ss.title == post_data['title'] tt = self.uu.get_by_title(self.wiki_title) assert tt.title == post_data['title'] def test_get_by_title2(self): '''Test Wiki title with SPACE''' post_data = { 'title': ' ' + self.wiki_title + ' ', 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', } uid = self.uu.insert_data(post_data) ss = self.uu.get_by_uid(uid) assert ss.title == self.wiki_title tt = self.uu.get_by_title(self.wiki_title) assert tt.title == post_data['title'].strip() def test_upate_by_view_count(self): post_data = { 'title': self.wiki_title, 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', } if self.uu.insert_data(post_data): rec = self.uu.get_by_wiki(self.wiki_title) viewcount0 = rec.view_count assert viewcount0 == 2 for x in range(100): self.uu.update_view_count_by_uid(rec.uid) viewcount1 = self.uu.get_by_wiki(self.wiki_title).view_count assert viewcount1 == 103 def test_upate(self): assert True def tearDown(self): print("function teardown") tt = self.uu.get_by_wiki(self.wiki_title) if tt: self.uu.delete(tt.uid)
class WikiHandler(BaseHandler): def initialize(self): self.init() self.mwiki = MWiki() self.mwiki_hist = MWikiHist() self.kind = '1' def get(self, url_str=''): url_arr = self.parse_url(url_str) if url_str == 'recent': self.recent() elif url_arr[0] == 'ajax_count_plus': self.ajax_count_plus(url_arr[1]) elif url_str == 'refresh': self.refresh() elif url_arr[0] == 'edit': self.to_modify(url_arr[1]) elif len(url_arr) == 1: self.wiki(url_str) else: kwd = { 'info': '页面未找到', } self.render('html/404.html', kwd=kwd) def post(self, url_str=''): url_arr = self.parse_url(url_str) if url_arr[0] == 'edit': self.update(url_arr[1]) elif url_arr[0] == 'add': self.wikinsert() else: self.redirect('html/404.html') def recent(self): kwd = { 'pager': '', 'unescape': tornado.escape.xhtml_unescape, 'title': '最近文档', } self.render( 'doc/wiki/wiki_list.html', view=self.mwiki.query_recent(), format_date=tools.format_date, # cfg=cfg, # Todo: Should delete. kwd=kwd, userinfo=self.userinfo, ) def refresh(self): kwd = { 'pager': '', 'unescape': tornado.escape.xhtml_unescape, 'title': '最近文档', } self.render( 'doc/wiki/wiki_list.html', view=self.mwiki.query_dated(16), format_date=tools.format_date, kwd=kwd, # cfg=cfg, # Todo: Should delete userinfo=self.userinfo, ) def wiki(self, title): postinfo = self.mwiki.get_by_wiki(title) if postinfo: if postinfo.kind == self.kind: self.viewit(postinfo) else: return False else: self.to_add(title) @tornado.web.authenticated def update(self, uid): raw_data = self.mwiki.get_by_id(uid) if self.check_post_role( self.userinfo )['EDIT'] or raw_data.user_name == self.get_current_user(): pass else: return False post_data = self.get_post_data() post_data['user_name'] = self.get_current_user() self.mwiki.update(uid, post_data) self.mwiki_hist.insert_data(raw_data) self.redirect('/wiki/{0}'.format( tornado.escape.url_escape(post_data['title']))) @tornado.web.authenticated def to_modify(self, id_rec): wiki_rec = self.mwiki.get_by_id(id_rec) # 用户具有管理权限,或文章是用户自己发布的。 if self.check_post_role( self.userinfo )['EDIT'] or wiki_rec.user_name == self.get_current_user(): pass else: return False kwd = { 'pager': '', } self.render( 'doc/wiki/wiki_edit.html', kwd=kwd, unescape=tornado.escape.xhtml_unescape, dbrec=wiki_rec, # Deprecated. postinfo=wiki_rec, # cfg=cfg, # Todo: Should delete userinfo=self.userinfo, ) def viewit(self, view): kwd = { 'pager': '', 'editable': self.editable(), } self.render( 'doc/wiki/wiki_view.html', view=view, # Deprecated postinfo=view, unescape=tornado.escape.xhtml_unescape, kwd=kwd, userinfo=self.userinfo, # cfg=cfg, # Todo: Should delete ) def ajax_count_plus(self, slug): output = { 'status': 1 if self.mwiki.update_view_count(slug) else 0, } return json.dump(output, self) def to_add(self, title): kwd = { 'title': title, 'pager': '', } if self.userinfo and self.userinfo.role[0] > '0': tmpl = 'doc/wiki/wiki_add.html' else: tmpl = 'doc/wiki/wiki_login.html' self.render( tmpl, kwd=kwd, # cfg= cfg, # Todo: should delete userinfo=self.userinfo, ) @tornado.web.authenticated def wikinsert(self): if self.userinfo.role[0] > '0': pass else: return False post_data = self.get_post_data() post_data['user_name'] = self.get_current_user() if self.mwiki.get_by_wiki(post_data['title']): pass else: self.mwiki.insert_data(post_data) self.redirect('/wiki/{0}'.format( tornado.escape.url_escape(post_data['title'])))