class TestPost(): def setup(self): print('setup 方法执行于本类中每条用例之前') self.uu = MPost() self.raw_count = self.uu.get_counts() self.post_title = 'ccc' self.uid = tools.get_uu4d() def test_insert(self): raw_count = self.uu.get_counts() post_data = { 'title': self.post_title, 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', 'view_count': 1, 'logo': '/static/', 'keywords': 'sdf', } self.uu.insert_data(self.uid, post_data) new_count = self.uu.get_counts() tt = self.uu.get_by_uid(self.uid) assert tt.title == post_data['title'] assert tt.cnt_md == tornado.escape.xhtml_unescape(post_data['cnt_md']) assert tt.cnt_html == tools.markdown2html(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', 'view_count': 1, 'logo': '/static/', 'keywords': 'sdf', } uu = self.uu.insert_data(self.uid, post_data) assert uu == False post_data = { 'title': '1', 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', 'view_count': 1, 'logo': '/static/', 'keywords': 'sdf', } uu = self.uu.insert_data(self.uid, post_data) assert uu == False post_data = { 'title': '天', 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', 'view_count': 1, 'logo': '/static/', 'keywords': 'sdf', } uu = self.uu.insert_data(self.uid, post_data) assert uu == False def test_get_by_title(self): post_data = { 'title': self.post_title, 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', 'view_count': 1, 'logo': '/static/', 'keywords': 'sdf', } uid = self.uu.insert_data(self.uid, post_data) ss = self.uu.get_by_uid(uid) assert ss.title == post_data['title'] def test_get_by_title2(self): '''Test Wiki title with SPACE''' post_data = { 'title': ' ' + self.post_title + ' ', 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', 'view_count': 1, 'logo': '/static/', 'keywords': 'sdf', } uid = self.uu.insert_data(self.uid, post_data) ss = self.uu.get_by_uid(uid) assert ss.title == self.post_title def test_update_by_uid(self): uid = self.uid post_data = { 'title': 'a123sdf', 'cnt_md': '1212sadf', 'user_name': 'asdf', 'logo': 'asqwef', 'keywords': 'aseef', } self.uu.insert_data(uid, post_data) new_count = self.uu.get_counts() #assert self.raw_count + 1 == new_count post_data2 = { 'title': 'a123sdf', 'cnt_md': '1212sadf', 'user_name': 'asdf', 'logo': '1111asqwef', 'keywords': '111aseef', } self.uu.update(uid, post_data2) new_count = self.uu.get_counts() #assert self.raw_count + 1 == new_count tt = self.uu.get_by_uid(uid) #assert tt.title != post_data['title'][0] #assert tt.cnt_md != post_data['cnt_md'][0] #assert tt.user_name != int(post_data['user_name'][0]) #assert tt.logo != post_data['logo'][0] #assert tt.keywords != post_data['keywords'][0] # #assert tt.title == post_data['title'][0] #assert tt.cnt_md == post_data['cnt_md'][0] #assert tt.user_name == int(post_data['user_name'][0]) #assert tt.logo == post_data['logo'][0] #assert tt.keywords == post_data['keywords'][0] def test_query_cat_random(self): self.uu.query_cat_random('') assert True def test_query_recent(self): self.uu.query_recent() assert True def test_query_all(self): self.uu.query_all() assert True def test_query_keywords_empty(self): self.uu.query_keywords_empty() assert True def test_query_dated(self): self.uu.query_dated() assert True def test_query_most_pic(self): self.uu.query_most_pic(3) assert True def test_query_cat_recent(self): self.uu.query_cat_recent(3, 3) assert True def test_query_most(self): self.uu.query_most() assert True def test_update_keywords(self): self.uu.update_keywords(self.uid, 'adf') assert True def test_update_view_count_by_uid(self): uid = tools.get_uu4d() post_data = { 'title': self.post_title, 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', 'view_count': 1, 'logo': '/static/', 'keywords': 'sdf', } self.uu.insert_data(uid, post_data) rec = self.uu.get_by_uid(uid) viewcount0 = rec.view_count assert viewcount0 == 1 for x in range(100): self.uu.update_view_count_by_uid(rec.uid) viewcount1 = self.uu.get_by_uid(uid).view_count assert viewcount1 == 101 def test_upate(self): assert True def tearDown(self): print("function teardown") tt = self.uu.get_by_uid(self.uid) if tt: self.uu.delete(tt.uid)
class PostHandler(BaseHandler): def initialize(self): self.init() self.mpost = MPost() self.mcat = MCategory() self.cats = self.mcat.query_all() self.mpost_hist = MPostHist() self.mpost2catalog = MPost2Catalog() self.mpost2label = MPost2Label() self.mrel = MRelation() self.tmpl_dir = 'doc' self.kind = '1' self.tmpl_router = 'post' def get(self, url_str=''): url_arr = self.parse_url(url_str) if url_str == '': self.recent() elif len(url_arr) == 1 and url_str.endswith('.html'): self.view_or_add(url_str.split('.')[0]) elif url_str == 'add_document': self.to_add_document() elif url_arr[0] == 'add_document': self.to_add_document() elif url_str == 'recent': self.recent() elif url_str == 'refresh': self.refresh() elif url_arr[0] in ['modify', 'edit']: self.to_modify(url_arr[1]) elif url_arr[0] == 'delete': self.delete(url_arr[1]) elif url_arr[0] == 'ajax_count_plus': self.ajax_count_plus(url_arr[1]) elif len(url_arr) == 1: self.view_or_add(url_str) else: kwd = { 'info': '页面未找到', } self.render( 'html/404.html', kwd=kwd, userinfo=self.userinfo, ) def post(self, url_str=''): if url_str == '': return url_arr = self.parse_url(url_str) if url_arr[0] in ['modify', 'edit']: self.update(url_arr[1]) elif url_arr[0] in ['add_document', '_add']: self.user_add_post() elif len(url_arr) == 1 and url_str.endswith('.html'): self.add_post(url_str) else: self.redirect('html/404.html') def ajax_count_plus(self, uid): output = { 'status': 1 if self.mpost.update_view_count_by_uid(uid) else 0, } return json.dump(output, self) def recent(self, with_catalog=True, with_date=True): kwd = { 'pager': '', 'unescape': tornado.escape.xhtml_unescape, 'title': '最近文档', 'with_catalog': with_catalog, 'with_date': with_date, } self.render( '{1}/{0}/post_list.html'.format(self.tmpl_router, self.tmpl_dir), kwd=kwd, view=self.mpost.query_recent(), view_all=self.mpost.query_all(), format_date=tools.format_date, userinfo=self.userinfo, cfg=config.cfg, ) @tornado.web.authenticated def __could_edit(self, postid): post_rec = self.mpost.get_by_uid(postid) if not post_rec: return False if self.check_post_role( self.userinfo )['EDIT'] or post_rec.user_name == self.userinfo.user_name: return True else: return False def refresh(self): kwd = { 'pager': '', 'title': '最近文档', } self.render( 'doc/post/post_list.html', kwd=kwd, userinfo=self.userinfo, view=self.mpost.query_dated(10), format_date=tools.format_date, unescape=tornado.escape.xhtml_unescape, cfg=config.cfg, ) # def get_random(self): # return self.mpost.query_random() def view_or_add(self, uid): if self.mpost.get_by_id(uid): self.view_post(uid) else: self.to_add(uid) @tornado.web.authenticated def to_add_document(self, ): if self.check_post_role(self.userinfo)['ADD']: pass else: return False kwd = { 'pager': '', 'cats': self.cats, 'uid': '', } self.render( '{1}/{0}/post_add.html'.format(self.tmpl_router, self.tmpl_dir), kwd=kwd, tag_infos=self.mcat.query_all(), userinfo=self.userinfo, cfg=config.cfg, ) @tornado.web.authenticated def to_add(self, uid): if self.check_post_role(self.userinfo)['ADD']: pass else: return False kwd = { 'cats': self.cats, 'uid': uid, 'pager': '', } self.render( 'doc/post/post_add.html', kwd=kwd, tag_infos=self.mcat.query_all(), cfg=config.cfg, userinfo=self.userinfo, ) @tornado.web.authenticated def update(self, uid): if self.__could_edit(uid): pass else: return False post_data = self.get_post_data() post_data['user_name'] = self.get_current_user() is_update_time = True # if post_data['is_update_time'][0] == '1' else False self.mpost_hist.insert_data(self.mpost.get_by_id(uid)) self.mpost.update(uid, post_data, update_time=is_update_time) self.update_catalog(uid) self.update_tag(uid) self.redirect('/post/{0}.html'.format(uid)) @tornado.web.authenticated def update_tag(self, signature): current_tag_infos = self.mpost2label.get_by_id(signature, kind=self.kind + '1') post_data = self.get_post_data() if 'tags' in post_data: pass else: return False print('tags: {0}'.format(post_data['tags'])) tags_arr = [x.strip() for x in post_data['tags'].split(',')] for tag_name in tags_arr: if tag_name == '': pass else: self.mpost2label.add_record(signature, tag_name, 1, kind=self.kind + '1') for cur_info in current_tag_infos: print(cur_info.tag.name) if cur_info.tag.name in tags_arr: pass else: self.mpost2label.remove_relation(signature, cur_info.tag) @tornado.web.authenticated def update_catalog(self, uid): post_data = self.get_post_data() current_infos = self.mpost2catalog.query_by_entity_uid(uid, kind=self.kind + '0') new_tag_arr = [] # HTML中预定义的 def_cate_arr = ['gcat{0}'.format(x) for x in range(10)] # todo: next line should be deleted. keep here for historical reason. def_cate_arr.append('def_cat_uid') for key in def_cate_arr: if key in post_data: pass else: continue print('a' * 4) print(post_data[key]) if post_data[key] == '' or post_data[key] == '0': continue # if len(post_data[key]) != 4: # continue print(post_data[key]) print(new_tag_arr) # 有可能选重复了。保留前面的 if post_data[key] in new_tag_arr: continue new_tag_arr.append(post_data[key] + ' ' * (4 - len(post_data[key]))) for idx, val in enumerate(new_tag_arr): self.mpost2catalog.add_record(uid, val, idx) # 对原来的进行处理,如果不在现有中,则删除 for cur_info in current_infos: if str(cur_info.tag.uid).strip() not in new_tag_arr: self.mpost2catalog.remove_relation(uid, cur_info.tag) @tornado.web.authenticated def to_modify(self, id_rec): if self.__could_edit(id_rec): pass else: return False kwd = { 'pager': '', 'cats': self.cats, } self.render( 'doc/post/post_edit.html', kwd=kwd, unescape=tornado.escape.xhtml_unescape, tag_infos=self.mcat.query_all(kind=constant['cate_post']), app2label_info=self.mpost2label.get_by_id( id_rec, kind=constant['tag_post']), app2tag_info=self.mpost2catalog.query_by_entity_uid( id_rec, kind=constant['cate_post']), dbrec=self.mpost.get_by_id(id_rec), userinfo=self.userinfo, cfg=config.cfg, ) def get_cat_str(self, cats): cat_arr = cats.split(',') out_str = '' for xx in self.cats: if str(xx.uid) in cat_arr: tmp_str = '''<li><a href="/category/{0}" style="margin:10px auto;"> {1} </a></li> '''.format(xx.slug, tornado.escape.xhtml_escape(xx.name)) out_str += tmp_str return (out_str) def get_cat_name(self, id_cat): for x in self.cats: if x['id_cat'] == id_cat: return (x['name']) def __gen_last_current_relation(self, post_id): ''' Generate the relation for the post and last post viewed. :param post_id: :return: ''' last_post_id = self.get_secure_cookie('last_post_uid') if last_post_id: last_post_id = last_post_id.decode('utf-8') self.set_secure_cookie('last_post_uid', post_id) if last_post_id and self.mpost.get_by_id(last_post_id): self.add_relation(last_post_id, post_id) def view_post(self, post_id): self.__gen_last_current_relation(post_id) cats = self.mpost2catalog.query_by_entity_uid(post_id) # replys = self.mpost2reply.get_by_id(post_id) tag_info = self.mpost2label.get_by_id(post_id) rec = self.mpost.get_by_id(post_id) if not rec: kwd = { 'info': '您要查看的页面不存在。', } self.render('html/404.html', kwd=kwd, userinfo=self.userinfo) return False if cats.count() == 0: cat_id = '' else: cat_id = cats.get().tag kwd = {'pager': '', 'editable': self.editable(), 'cat_id': cat_id} rel_recs = self.mrel.get_app_relations(rec.uid, 4) rand_recs = self.mpost.query_random(4 - rel_recs.count() + 2) self.render( 'doc/post/post_view.html', view=rec, postinfo=rec, unescape=tornado.escape.xhtml_unescape, kwd=kwd, userinfo=self.userinfo, tag_info=tag_info, relations=rel_recs, rand_recs=rand_recs, replys=[], cfg=config.cfg, ) def add_relation(self, f_uid, t_uid): if self.mpost.get_by_id(t_uid) is False: return False if f_uid == t_uid: ''' 关联其本身 ''' return False # 双向关联,但权重不一样. self.mrel.add_relation(f_uid, t_uid, 2) self.mrel.add_relation(t_uid, f_uid, 1) return True @tornado.web.authenticated def add_post(self, url_str): url_arr = url_str.split('.') if len(url_arr) == 2: id_post = url_arr[0] if len(id_post) == 5: pass else: return False else: return False if self.check_post_role(self.userinfo)['ADD']: pass else: return False post_data = self.get_post_data() post_data['user_name'] = self.userinfo.user_name cur_post_rec = self.mpost.get_by_id(id_post) if cur_post_rec is None: uid = self.mpost.insert_data(id_post, post_data) self.update_tag(uid) self.update_catalog(uid) self.redirect('/post/{0}.html'.format(id_post)) @tornado.web.authenticated def user_add_post(self): if self.check_post_role(self.userinfo)['ADD']: pass else: return False post_data = self.get_post_data() if not ('title' in post_data): self.set_status(400) return False else: pass post_data['user_name'] = self.get_current_user() cur_uid = tools.get_uu5d() while self.mpost.get_by_id(cur_uid): cur_uid = tools.get_uu5d() uid = self.mpost.insert_data(cur_uid, post_data) self.update_tag(uid) self.update_catalog(uid) self.redirect('/post/{0}.html'.format(cur_uid)) @tornado.web.authenticated def delete(self, del_id): if self.check_post_role(self.userinfo)['DELETE']: pass else: return False is_deleted = self.mpost.delete(del_id) if is_deleted: self.redirect('/post/recent') else: return False
class TestPost(): def setup(self): print('setup 方法执行于本类中每条用例之前') self.post = MPost() self.user = MUser() self.reply = MReply() self.post_title = 'ccc' self.username = '******' self.uid = tools.get_uu4d() def test_insert_post(self): raw_count = self.post.get_counts() post_data = { 'title': self.post_title, 'cnt_md': '## adslkfjasdf\n lasdfkjsadf', 'user_name': 'Tome', 'view_count': 1, 'logo': '/static/', 'keywords': 'sdf', } self.post.insert_data(self.uid, post_data) tt = self.post.get_by_uid(self.uid) def test_insert_user(self): post_data = { 'user_name': self.username, 'user_pass': '******', 'user_email': '*****@*****.**', } tt = self.user.insert_data(post_data) print('=' * 20) print(tt) assert tt['success'] == True def test_insert_reply(self): post_data = { 'user_name': self.username, 'user_pass': '******', 'user_email': '*****@*****.**', } tt = self.user.insert_data(post_data) # u_id = self.user.get_by_name(self.username) # # # self.userid = u_id.uid # # print("*" * 50) # print(self.userid) # print("*" * 50) # post_data = { # 'user_name':[self.username], # 'create_user_id':[self.userid], # 'timestamp':['2'], # 'date':['1'], # 'cnt_md':['###adfafasf/sdf'], # 'cnt_html':['###adfafasf/sdf'], # 'vote':0, # } def test_upate(self): assert True def tearDown(self): print("function teardown") tt = self.post.get_by_uid(self.uid) if tt: self.post.delete(tt.uid) self.user.delete_by_user_name(self.username)