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()