def update(self, uid):
'''
Update the link
:param uid:
:return:
'''
if self.userinfo.role[1] >= '3':
pass
else:
return False
post_data = self.get_post_data()
post_data['user_name'] = self.get_current_user()
if self.is_p == True:
if MLink.update(uid, post_data):
output = {
'addinfo ': 1,
}
else:
output = {
'addinfo ': 0,
}
return json.dump(output, self)
else:
if MLink.update(uid, post_data):
self.redirect('/link/list')
def p_user_add_link(self):
'''
user add link.
'''
if self.check_post_role()['ADD']:
pass
else:
return False
post_data = self.get_post_data()
post_data['user_name'] = self.get_current_user()
cur_uid = tools.get_uudd(2)
while MLink.get_by_uid(cur_uid):
cur_uid = tools.get_uudd(2)
if MLink.create_link(cur_uid, post_data):
output = {
'addinfo ': 1,
}
else:
output = {
'addinfo ': 0,
}
return json.dump(output, self)
def test_get_by_uid(self):
a = MLink.get_by_uid(self.id)
assert a == None
self.add_message()
a = MLink.get_by_uid(self.id)
assert a.uid == self.id
self.tearDown()
def add_message(self, **kwargs):
uid = self.id
post_data = {
'name': kwargs.get('name', 'asdf'),
'link': kwargs.get('link', 'sadf'),
'order': kwargs.get('order', '1'),
'logo': kwargs.get('logo', 'asf'),
}
MLink.create_link(uid, post_data)
def test_mlink_insert():
''' Test mlink insert '''
# assert 0
uu = MLink()
uid = tools.get_uu4d()
raw_count = uu.get_counts()
post_data = {
'name': 'asdf',
'link': 'sadf',
'order': '1',
'logo': 'asf',
}
uu.insert_data(uid, post_data)
new_count = uu.get_counts()
tt = uu.get_by_uid(uid)
# assert
assert tt.name == post_data['name']
assert tt.link == post_data['link']
assert tt.order == int(post_data['order'])
assert tt.logo == post_data['logo']
assert raw_count + 1 == new_count
uu.delete(uid)
def delete(self, del_id):
if self.check_post_role()['DELETE']:
pass
else:
return False
if self.is_p == True:
if MLink.delete(del_id):
output = {'del_link': 1}
else:
output = {'del_link': 0}
return json.dump(output, self)
else:
is_deleted = MLink.delete(del_id)
if is_deleted:
self.redirect('/link/list')
def test_query_all(self):
a = MLink.query_all()
tf = True
for i in a:
if i.uid == self.id:
tf = False
assert tf
self.add_message()
a = MLink.query_all()
tf = False
for i in a:
if i.uid == self.id:
tf = True
self.tearDown()
assert tf
class link_list(tornado.web.UIModule):
def render(self, num=10):
self.mlink = MLink()
recs = self.mlink.query_link(num)
return self.render_string('modules/post/link_list.html',
recs=recs,
)
class IndexHandler(BaseHandler):
def initialize(self):
self.init()
self.mpost = MPost()
self.mcat = MCategory()
self.mpage = MPage()
self.mlink = MLink()
def get_current_user(self):
return self.get_secure_cookie("user")
def get(self, input=''):
if input == '':
self.index()
else:
self.render('html/404.html', kwd={}, userinfo=self.userinfo)
def index(self):
cstr = tools.get_uuid()
self.set_cookie('user_pass', cstr)
kwd = {'cookie_str': cstr}
self.render(
'index/index.html',
userinfo=self.userinfo,
catalog_info=self.mcat.query_all(by_order=True),
link=self.mlink.query_all(),
unescape=tornado.escape.xhtml_unescape,
cfg=config.cfg,
view=self.mpost.query_most_pic(20),
kwd=kwd,
)
def user_add_link(self):
if self.check_post_role()['ADD']:
pass
else:
return False
post_data = self.get_post_data()
post_data['user_name'] = self.get_current_user()
cur_uid = tools.get_uudd(2)
while MLink.get_by_uid(cur_uid):
cur_uid = tools.get_uudd(2)
MLink.create_link(cur_uid, post_data)
self.redirect('/link/list')
def test_create_link(self):
uid = self.id
post_data = {
'name': 'asdf',
'link': 'sadf',
'order': '1',
'logo': 'asf',
}
MLink.create_link(uid, post_data)
new_count = MLink.get_counts()
tt = MLink.get_by_uid(uid)
assert tt.name == post_data['name']
assert tt.link == post_data['link']
assert tt.order == int(post_data['order'])
assert tt.logo == post_data['logo']
assert self.raw_count + 1 == new_count
def delete(self, del_id):
if self.check_post_role()['DELETE']:
pass
else:
return False
if MLink.delete(del_id):
output = {'del_link': 1}
else:
output = {'del_link': 0}
return json.dump(output, self)
def recent(self):
kwd = {
'pager': '',
'unescape': tornado.escape.xhtml_unescape,
'title': '最近文档',
}
self.render('misc/link/link_list.html',
kwd=kwd,
view=MLink.query_link(10),
format_date=tools.format_date,
userinfo=self.userinfo)
def recent(self):
kwd = {
'pager': '',
'unescape': tornado.escape.xhtml_unescape,
'title': '最近文档',
}
if self.is_p == True:
self.render('admin/link_ajax/link_list.html',
kwd=kwd,
view=MLink.query_link(20),
format_date=tools.format_date,
userinfo=self.userinfo)
else:
self.render('misc/link/link_list.html',
kwd=kwd,
view=MLink.query_link(20),
format_date=tools.format_date,
userinfo=self.userinfo)
def to_modify(self, id_rec):
if self.userinfo.role[1] >= '3':
pass
else:
return False
a = MLink.get_by_uid(id_rec)
self.render('misc/link/link_edit.html',
kwd={},
unescape=tornado.escape.xhtml_unescape,
dbrec=a,
userinfo=self.userinfo)
def recent(self):
'''
Recent links.
'''
kwd = {
'pager': '',
'title': '最近文档',
}
if self.is_p:
self.render('admin/link_ajax/link_list.html',
kwd=kwd,
view=MLink.query_link(20),
format_date=tools.format_date,
userinfo=self.userinfo)
else:
self.render('misc/link/link_list.html',
kwd=kwd,
view=MLink.query_link(20),
format_date=tools.format_date,
userinfo=self.userinfo)
def index(self):
'''
Index funtion.
'''
self.render('index/index.html',
userinfo=self.userinfo,
catalog_info=MCategory.query_all(by_order=True),
link=MLink.query_all(),
unescape=tornado.escape.xhtml_unescape,
cfg=CMS_CFG,
view=MPost.query_most_pic(20),
kwd={}, )
def to_modify(self, uid):
'''
Try to edit the link.
'''
if self.userinfo.role[1] >= '3':
pass
else:
return False
self.render('misc/link/link_edit.html',
kwd={},
postinfo=MLink.get_by_uid(uid),
userinfo=self.userinfo)
def test_upate(self):
post_data = {
'name': 'fffffff',
'link': '85tgr4ggbf',
'order': '13',
'logo': 'fdef',
}
self.add_message(**post_data)
new_count = MLink.get_counts()
assert self.raw_count + 1 <= new_count
post_data2 = {
'name': 'asdlkjf',
'link': 'sakljdf',
'order': '12',
'logo': 'asfa',
}
MLink.update(self.id, post_data2)
new_count = MLink.get_counts()
assert self.raw_count + 1 <= new_count
tt = MLink.get_by_uid(self.id)
assert tt.name != post_data['name']
assert tt.link != post_data['link']
assert tt.order != int(post_data['order'])
assert tt.logo != post_data['logo']
assert tt.name == post_data2['name']
assert tt.link == post_data2['link']
assert tt.order == int(post_data2['order'])
assert tt.logo == post_data2['logo']
self.tearDown()
def to_modify(self, uid):
'''
Try to edit the link
:param uid:
:return:
'''
if self.userinfo.role[1] >= '3':
pass
else:
return False
self.render('misc/link/link_edit.html',
kwd={},
unescape=tornado.escape.xhtml_unescape,
dbrec=MLink.get_by_uid(uid),
userinfo=self.userinfo)
def update(self, uid):
if self.userinfo.role[1] >= '3':
pass
else:
return False
post_data = self.get_post_data()
post_data['user_name'] = self.get_current_user()
if MLink.update(uid, post_data):
output = {
'addinfo ': 1,
}
else:
output = {
'addinfo ': 0,
}
return json.dump(output, self)
def viewit(self, post_id):
rec = MLink.get_by_uid(post_id)
if not rec:
kwd = {'info': '您要找的分类不存在。'}
self.render('misc/html/404.html', kwd=kwd)
return False
kwd = {
'pager': '',
'editable': self.editable(),
}
self.render('misc/link/link_view.html',
view=rec,
unescape=tornado.escape.xhtml_unescape,
kwd=kwd,
userinfo=self.userinfo,
cfg=CMS_CFG, )
def test_upate(self):
uid = self.id
post_data = {
'name': 'asdf',
'link': 'sadf',
'order': '1',
'logo': 'asf',
}
MLink.create_link(uid, post_data)
new_count = MLink.get_counts()
assert self.raw_count + 1 == new_count
post_data2 = {
'name': 'asdlkjf',
'link': 'sakljdf',
'order': '13',
'logo': 'asfa',
}
MLink.update(uid, post_data2)
new_count = MLink.get_counts()
assert self.raw_count + 1 == new_count
tt = MLink.get_by_uid(uid)
assert tt.name != post_data['name']
assert tt.link != post_data['link']
assert tt.order != int(post_data['order'])
assert tt.logo != post_data['logo']
assert tt.name == post_data2['name']
assert tt.link == post_data2['link']
assert tt.order == int(post_data2['order'])
assert tt.logo == post_data2['logo']
def setup(self):
self.id = tools.get_uu4d()
self.raw_count = MLink.get_counts()
print(self.raw_count)
def tearDown(self):
print("function teardown")
tt = MLink.get_by_uid(self.id)
if tt:
MLink.delete(tt.uid)
def test_get_counts(self):
a = MLink.get_counts()
self.add_message()
b = MLink.get_counts()
assert a + 1 <= b
self.tearDown()
def initialize(self):
self.init()
self.mpost = MPost()
self.mcat = MCategory()
self.mpage = MPage()
self.mlink = MLink()
def render(self, num=10):
return self.render_string('modules/post/link_list.html',
recs=MLink.query_link(num))
def render(self, *args, **kwargs):
num = kwargs['num'] if 'num' in kwargs else 10
return self.render_string('modules/post/link_list.html',
recs=MLink.query_link(num))
def initialize(self):
self.init()
self.user_name = self.get_current_user()
self.tmpl_dir = 'admin'
self.tmpl_router = 'link_ajax'
self.mlink = MLink()
