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_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 tearDown(self):
MCategory.delete(self.tag_id)
MPost.delete(self.post_id2)
MPost.delete(self.post_id)
MPost2Catalog.remove_relation(self.post_id, self.tag_id)
MRelation.delete(self.uid)
def test_update_relation(self):
self.add_tag()
self.add_2post()
self.add_rela()
MRelation.update_relation(self.post_id, self.post_id2, weight=2)
aa = MHelper.get_by_uid(TabRel, self.uid)
assert aa.count >= 2 + 1
self.tearDown()
def test_delete(self):
self.add_tag()
self.add_2post()
self.add_rela()
aa = MHelper.get_by_uid(TabRel, self.uid)
assert aa.post_f_id == self.post_id
MRelation.delete(self.uid)
aa = MHelper.get_by_uid(TabRel, self.uid)
assert aa == None
self.tearDown()
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.kind = '1'
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 render(self, *args, **kwargs):
# uid = args[0]
# num = args[1] if len(args) > 1 else 6
# kind = kwargs['kind'] if 'kind' in kwargs else '1'
# num = kwargs['num'] if 'num' in kwargs else num
uid = kwargs.get('uid', args[0])
num = kwargs.get('num', args[1] if len(args) > 1 else 6)
kind = kwargs.get('kind', '1')
kwd = {
'app_f': 'post',
'app_t': 'info',
'uid': uid,
}
rel_recs = MRelation.get_app_relations(uid, num, kind=kind).objects()
rand_recs = MPost.query_random(num=num - rel_recs.count() + 2,
kind=kind)
return self.render_string(
'modules/info/relation_post2app.html',
relations=rel_recs,
rand_recs=rand_recs,
kwd=kwd,
)
def tearDown(self):
print("function teardown")
tt = self.uu.get_by_uid(self.app_f)
if tt:
MPost2Catalog.remove_relation(self.app_t, self.tag_id)
MPost2Catalog.remove_relation(self.app_f, self.tag_id)
MCategory.delete(self.tag_id)
MPost.delete(self.app_f)
MPost.delete(self.app_t)
MRelation.delete(self.uid)
tt = MLabel.get_by_slug(self.slug)
if tt:
MLabel.delete(tt.uid)
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_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()
def test_get_app_relations(self):
self.add_tag()
self.add_2post()
self.add_rela()
aa = MRelation.get_app_relations(self.post_id)
tf = False
for i in aa:
print(i.post_id)
if i.post_id == self.post_id2:
tf = True
self.tearDown()
assert tf
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.mpost2reply = MPost2Reply()
self.mpost2label = MPost2Label()
self.mrel = MRelation()
self.tmpl_dir = 'admin'
self.tmpl_router = 'post_ajax'
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 render(self, *args, **kwargs):
uid = args[0]
num = args[1]
kwd = {
'app_f': 'info',
'app_t': 'post',
'uid': uid,
}
rel_recs = MRelation.get_app_relations(uid, num, kind='1').naive()
rand_recs = MPost.query_random(num=num - rel_recs.count() + 2, kind='1')
return self.render_string('modules/info/relation_app2post.html',
relations=rel_recs,
rand_recs=rand_recs,
kwd=kwd)
def fetch_additional_posts(self, uid):
'''
fetch the rel_recs, and random recs when view the post.
'''
cats = MPost2Catalog.query_by_entity_uid(uid, kind=self.kind)
cat_uid_arr = []
for cat_rec in cats:
cat_uid = cat_rec.tag_id
cat_uid_arr.append(cat_uid)
logger.info('info category: {0}'.format(cat_uid_arr))
rel_recs = MRelation.get_app_relations(uid, 8, kind=self.kind).objects()
logger.info('rel_recs count: {0}'.format(rel_recs.count()))
if cat_uid_arr:
rand_recs = MPost.query_cat_random(cat_uid_arr[0], limit=4 - rel_recs.count() + 4)
else:
rand_recs = MPost.query_random(num=4 - rel_recs.count() + 4, kind=self.kind)
return rand_recs, rel_recs
def render(
self,
uid,
num,
):
kwd = {
'app_f': 'post',
'app_t': 'info',
'uid': uid,
}
rel_recs = MRelation.get_app_relations(uid, num, kind='9')
rand_recs = MPost.query_random(num=num - rel_recs.count() + 2,
kind='9')
return self.render_string(
'modules/info/relation_post2app.html',
relations=rel_recs,
rand_recs=rand_recs,
kwd=kwd,
)
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 test_add_relation(self):
MRelation.add_relation(self.app_f, self.app_t)
assert True
def test_update_relation(self):
MRelation.update_relation(self.app_f, self.app_t)
assert True
def test_get_app_relations(self):
MRelation.get_app_relations(self.app_f)
assert True
def add_rela(self):
aa = MRelation.add_relation(self.post_id, self.post_id2)
self.uid = aa
def test_delete(self):
MRelation.delete(self.app_f)
assert True
def Test():
assert MRelation()
assert MRelation()
assert MRelation()
def tearDown(self):
MRelation.delete(self.app_id)
def test_delete(self):
app_id = self.app_id
MRelation.delete(app_id)
assert True
def test_get_app_relations(self):
app_id = self.app_id
MRelation.get_app_relations(app_id)
assert True
def test_update_relation(self):
app_f = self.app_id
app_t = '12345'
MRelation.update_relation(app_f, app_t)
assert True
def test_add_relation(self):
app_f = self.app_id
app_t = '12345'
MRelation.add_relation(app_f, app_t)
assert True
