def test_create_post_history(self):
self.tearDown()
p_d = {
'title': 'qqqii',
'cnt_md': 'qwqwqw',
'time_create': '1999',
'time_update': '2019',
'user_name': 'max',
'view_count': '1',
'logo': 'opps',
'memo': '',
'order': '1',
'kind': '1',
'valid': 1,
}
MPost().add_meta(self.post_id, p_d)
aa = MPost.get_by_uid(self.post_id)
tf = MPostHist.create_post_history(aa, aa)
assert tf
His = MPostHist.query_by_postid(self.post_id)
self.uid = His[0].uid
assert His[0].cnt_md == p_d['cnt_md']
self.tearDown()
def test_query_by_postid(self):
p_t = {'cnt_md': 'bbrreedd', 'user_name': 'ggggbabybaby'}
self.addHis(**p_t)
aa = MPostHist.query_by_postid(self.post_id)
assert aa[0].cnt_md == p_t['cnt_md']
assert aa[0].user_name == p_t['user_name']
self.tearDown()
def view(self, uid):
postinfo = MPost.get_by_uid(uid)
if postinfo:
pass
else:
return
hist_recs = MPostHist.query_by_postid(uid, limit=5)
html_diff_arr = []
for hist_rec in hist_recs:
if hist_rec:
infobox = diff_table(hist_rec.cnt_md, postinfo.cnt_md)
else:
infobox = ''
html_diff_arr.append({
'hist_uid': hist_rec.uid,
'html_diff': infobox
})
self.render('man_info/post_man_view.html',
userinfo=self.userinfo,
unescape=tornado.escape.xhtml_unescape,
view=postinfo,
postinfo=postinfo,
html_diff_arr=html_diff_arr,
router=router_post[postinfo.kind])
def view(self, uid):
postinfo = MPost.get_by_uid(uid)
if postinfo:
pass
else:
return
hist_recs = MPostHist.query_by_postid(uid, limit=5)
html_diff_arr = []
for hist_rec in hist_recs:
if hist_rec:
infobox = diff_table(hist_rec.cnt_md, postinfo.cnt_md)
hist_user = hist_rec.user_name
hist_time = hist_rec.time_update
hist_words_num = len((hist_rec.cnt_md).strip())
post_words_num = len((postinfo.cnt_md).strip())
up_words_num = post_words_num - hist_words_num
else:
infobox = ''
hist_user = ''
hist_time = ''
up_words_num = ''
html_diff_arr.append({
'hist_uid': hist_rec.uid,
'html_diff': infobox,
'hist_user': hist_user,
'hist_time': hist_time,
'up_words_num': up_words_num
})
kwd = {}
self.render('man_info/post_man_view.html',
userinfo=self.userinfo,
view=postinfo,
postinfo=postinfo,
html_diff_arr=html_diff_arr,
router=router_post[postinfo.kind],
kwd=kwd)
def addHis(self, **kwargs):
p_d = {
'title': kwargs.get('title', 'iiiii'),
'cnt_md': kwargs.get('cnt_md', 'grgr'),
'time_create': kwargs.get('time_create', '1992'),
'time_update': kwargs.get('time_update', '1996070600'),
'user_name': kwargs.get('user_name', 'yuanyuan'),
'view_count': kwargs.get('view_count', 1),
'logo': kwargs.get('logo', 'prprprprpr'),
'memo': kwargs.get('memo', ''),
'order': kwargs.get('order', '1'),
'keywords': kwargs.get('keywords', ''),
'extinfo': kwargs.get('extinfo', {}),
'kind': kwargs.get('kind', '1'),
'valid': kwargs.get('valid', 1),
}
MPost().add_meta(self.post_id, p_d)
aa = MPost.get_by_uid(self.post_id)
MPostHist.create_post_history(aa, aa)
His = MPostHist.query_by_postid(self.post_id)
self.uid = His[0].uid
class PostManHandler(BaseHandler):
def initialize(self):
self.init()
self.mpost = MPost()
self.mposthist = MPostHist()
def get(self, url_str=''):
url_arr = self.parse_url(url_str)
if url_arr[0] == 'view':
self.view(url_arr[1])
elif url_arr[0] == 'edit':
self.to_edit(url_arr[1])
elif url_arr[0] == 'restore':
self.restore(url_arr[1])
elif url_arr[0] == 'delete':
self.delete(url_arr[1])
else:
kwd = {
'info': '页面未找到',
}
self.render(
'html/404.html',
kwd=kwd,
userinfo=self.userinfo,
)
def post(self, url_str=''):
url_arr = self.parse_url(url_str)
if url_arr[0] == 'edit':
self.update(url_arr[1])
else:
self.redirect('html/404.html')
@tornado.web.authenticated
def update(self, uid):
if self.userinfo.role[0] > '0':
pass
else:
return False
post_data = self.get_post_data()
if self.userinfo:
post_data['user_name'] = self.userinfo.user_name
else:
post_data['user_name'] = ''
cur_info = self.mpost.get_by_id(uid)
self.mposthist.insert_data(cur_info)
self.mpost.update_cnt(uid, post_data)
self.redirect('/{0}/{1}'.format(router_post[cur_info.kind], uid))
@tornado.web.authenticated
def to_edit(self, postid):
if self.userinfo.role[0] > '0':
pass
else:
return False
post_rec = self.mpost.get_by_uid(postid)
self.render(
'man_post/post_man_edit.html',
userinfo=self.userinfo,
unescape=tornado.escape.xhtml_unescape,
postinfo=post_rec,
)
@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
@tornado.web.authenticated
def delete(self, uid):
if self.check_post_role(self.userinfo)['DELETE']:
pass
else:
return False
histinfo = self.mposthist.get_by_id(uid)
if histinfo:
pass
else:
return False
postinfo = self.mpost.get_by_id(histinfo.post_id)
self.mposthist.delete(uid)
self.redirect('/post_man/view/{0}'.format(postinfo.uid))
def view(self, uid):
postinfo = self.mpost.get_by_id(uid)
if postinfo:
pass
else:
return
hist_recs = self.mposthist.query_by_postid(uid, limit=5)
html_diff_arr = []
for hist_rec in hist_recs:
if hist_rec:
infobox = diff_table(hist_rec.cnt_md, postinfo.cnt_md)
else:
infobox = ''
html_diff_arr.append({
'hist_uid': hist_rec.uid,
'html_diff': infobox
})
self.render(
'man_post/post_man_view.html',
userinfo=self.userinfo,
unescape=tornado.escape.xhtml_unescape,
view=postinfo,
postinfo=postinfo,
html_diff_arr=html_diff_arr,
router=router_post[postinfo.kind],
)
@tornado.web.authenticated
def restore(self, hist_uid):
if self.check_post_role(self.userinfo)['ADMIN']:
pass
else:
return False
histinfo = self.mposthist.get_by_id(hist_uid)
if histinfo:
pass
else:
return False
postinfo = self.mpost.get_by_id(histinfo.post_id)
cur_cnt = tornado.escape.xhtml_unescape(postinfo.cnt_md)
old_cnt = tornado.escape.xhtml_unescape(histinfo.cnt_md)
self.mpost.update_cnt(histinfo.post_id, {
'cnt_md': old_cnt,
'user_name': self.userinfo.user_name
})
self.mposthist.update_cnt(histinfo.uid, {
'cnt_md': cur_cnt,
'user_name': postinfo.user_name
})
self.redirect('/{0}/{1}'.format(router_post[postinfo.kind],
postinfo.uid))
def test_query_by_postid(self):
MPostHist.query_by_postid(self.post_id)
assert True
