def add_relation(self, f_uid, t_uid): ''' Add the relation. And the from and to, should have different weight. :param f_uid: :param t_uid: :return: return True if the relation has been succesfully added. ''' if not MPost.get_by_uid(t_uid): return False if f_uid == t_uid: return False # 针对分类进行处理。只有落入相同分类的,才加1 f_cats = MPost2Catalog.query_by_entity_uid(f_uid) t_cats = MPost2Catalog.query_by_entity_uid(t_uid) flag = False for f_cat in f_cats: for t_cat in t_cats: if f_cat.tag == t_cat.tag: flag = True if flag: pass else: return False # 双向关联,但权重不一样. MRelation.add_relation(f_uid, t_uid, 2) MRelation.add_relation(t_uid, f_uid, 1) return True
def test_get_app_relations(self): self.tearDown() self.add_message() MRelation.add_relation(self.app_f, self.app_t) a = MRelation.get_app_relations(self.app_f, num=300) tf = False for i in a: if i.post_id == self.app_f: tf = True assert tf self.tearDown()
def add_relation(self, url_arr): ''' Add relationship. :param url_arr: :return: ''' if MPost.get_by_uid(url_arr[1]): pass else: return False last_post_id = self.get_secure_cookie('last_post_uid') if last_post_id: last_post_id = last_post_id.decode('utf-8') last_app_id = self.get_secure_cookie('use_app_uid') if last_app_id: last_app_id = last_app_id.decode('utf-8') if url_arr[0] == 'info': if last_post_id: MRelation.add_relation(last_post_id, url_arr[1], 2) MRelation.add_relation(url_arr[1], last_post_id, 1) if url_arr[0] == 'post': if last_app_id: MRelation.add_relation(last_app_id, url_arr[1], 2) MRelation.add_relation(url_arr[1], last_app_id, 1)
def test_update_relation(self): self.add_message() a=MRelation.add_relation(self.app_f, self.app_t) self.uid = a a =MRelation.update_relation(self.app_f, self.app_t) assert a!=False self.tearDown()
def test_get_app_relations(self, **kwargs): MPost.delete(self.uid) MPost.delete(self.uid2) # 添加post post_data = { 'title': 'test1', 'cnt_md': '## test', 'user_name': 'test', 'view_count': 1, 'logo': ' ', 'keywords': 'test', 'kind': '1' } uu1 = MPost.create_post(self.uid, post_data) post_data2 = { 'title': 'test2', 'cnt_md': '## test2', 'user_name': 'test2', 'view_count': 1, 'logo': ' ', 'keywords': 'test2', 'kind': '1' } uu2 = MPost.create_post(self.uid2, post_data2) # 添加category cat_data = { 'name': kwargs.get('name', 'category'), 'slug': kwargs.get('slug', 'slug1'), 'order': kwargs.get('order', '0'), 'kind': kwargs.get('kind1', '1'), 'pid': kwargs.get('pid', '0000'), } MCategory.add_or_update(self.tag_id, cat_data) MPost2Catalog.add_record(uu1, self.tag_id) MRelation.add_relation(uu1, uu2) MRelation.get_app_relations(uu1) assert True
def test_delete(self): a = MRelation.add_relation(self.app_f, self.app_t) MRelation.update_relation(self.app_f, self.app_t) self.uid = a MRelation.delete(self.uid) au = MRelation.update_relation(self.app_f, self.app_t) assert au == False self.tearDown()
def test_add_relation(self): self.setup() self.add_message() a=MRelation.add_relation(self.app_f, self.app_t) self.uid=a assert a self.tearDown()
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
def add_rela(self): aa = MRelation.add_relation(self.post_id, self.post_id2) self.uid = aa
def test_add_relation(self): app_f = self.app_id app_t = '12345' MRelation.add_relation(app_f, app_t) assert True
def test_add_relation(self): MRelation.add_relation(self.app_f, self.app_t) assert True