def render(self, num=10):
self.mlink = MLink()
recs = self.mlink.query_link(num)
return self.render_string(
'doc/modules/link_list.html',
recs=recs,
)
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'][0]
assert tt.link == post_data['link'][0]
assert tt.order == int(post_data['order'][0])
assert tt.logo == post_data['logo'][0]
assert raw_count + 1 == new_count
uu.delete(uid)
class link_list(tornado.web.UIModule):
def render(self, num=10):
self.mlink = MLink()
recs = self.mlink.query_link(num)
return self.render_string('tmpl_torlite/modules/link_list.html',
recs=recs,
)
class IndexHandler(BaseHandler):
def initialize(self):
self.init()
self.mpost = MPost()
self.mcat = MCatalog()
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,
)
class IndexHandler(BaseHandler):
def initialize(self):
self.init()
self.mpost = MPost()
self.mcat = MCatalog()
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, )
class link_list(tornado.web.UIModule):
def render(self, num=10):
self.mlink = MLink()
recs = self.mlink.query_link(num)
return self.render_string(
'doc/modules/link_list.html',
recs=recs,
)
def initialize(self):
self.init()
self.mlink = MLink()
self.tmpl_router = 'link'
class LinkHandler(BaseHandler):
def initialize(self):
self.init()
self.mlink = MLink()
self.tmpl_router = 'link'
def get(self, url_str=''):
url_arr = self.parse_url(url_str)
if len(url_arr) == 1 and url_str.endswith('.html'):
self.wiki(url_str.split('.')[0])
elif url_str == 'add_link':
self.to_add_link()
elif url_str == 'list':
self.recent()
elif url_str == 'refresh':
self.refresh()
elif url_arr[0] == 'modify':
self.to_modify(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=''):
if url_str == '':
return
url_arr = url_str.split('/')
if url_arr[0] == 'modify':
self.update(url_arr[1])
elif url_str == 'add_link':
self.user_add_link()
elif url_arr[0] == 'add_link':
self.p_user_add_link()
else:
self.redirect('html/404.html')
def recent(self):
kwd = {
'pager': '',
'unescape': tornado.escape.xhtml_unescape,
'title': '最近文档',
}
self.render('{0}/{1}/link_list.html'.format(self.tmpl_name, self.tmpl_router),
kwd=kwd,
view=self.mlink.query_recent(),
format_date=tools.format_date,
userinfo=self.userinfo,
)
def refresh(self):
kwd = {
'pager': '',
'title': '最近文档',
}
self.render('{0}/link/link_list.html'.format(self.tmpl_name),
kwd=kwd,
userinfo=self.userinfo,
view=self.mlink.query_dated(10),
format_date=tools.format_date,
unescape=tornado.escape.xhtml_unescape, )
def get_random(self):
return self.mlink.query_random()
def wiki(self, uid):
dbdate = self.mlink.get_by_id(uid)
if dbdate:
self.viewit(uid)
else:
self.to_add(uid)
def to_add_link(self, ):
kwd = {
'pager': '',
'uid': '',
}
self.render('{0}/{1}/link_add.html'.format(self.tmpl_name, self.tmpl_router),
topmenu='',
kwd=kwd,
userinfo=self.userinfo,
)
@tornado.web.authenticated
def to_add(self, uid):
kwd = {
'uid': uid,
'pager': '',
}
self.render('{0}/{1}/link_add.html'.format(self.tmpl_name, self.tmpl_router),
kwd=kwd,
)
@tornado.web.authenticated
def update(self, uid):
raw_data = self.mlink.get_by_id(uid)
if self.userinfo.privilege[2] == '1' or raw_data.user_name == self.get_current_user():
pass
else:
return False
post_data = {}
for key in self.request.arguments:
post_data[key] = self.get_arguments(key)
post_data['user_name'] = self.get_current_user()
if self.tmpl_router == "link":
self.mlink.update(uid, post_data)
self.redirect('/link/list'.format(uid))
else:
if self.mlink.update(uid, post_data):
output = {
'addinfo ': 1,
}
else:
output = {
'addinfo ': 0,
}
return json.dump(output, self)
@tornado.web.authenticated
def to_modify(self, id_rec):
a = self.mlink.get_by_id(id_rec)
# 用户具有管理权限，
# 或
# 文章是用户自己发布的。
if self.userinfo.privilege[2] == '1' or a.user_name == self.get_current_user():
pass
else:
return False
kwd = {
'pager': '',
}
self.render('{0}/{1}/link_edit.html'.format(self.tmpl_name, self.tmpl_router),
kwd=kwd,
unescape=tornado.escape.xhtml_unescape,
dbrec=a,
userinfo=self.userinfo,
)
@tornado.web.authenticated
def viewit(self, post_id):
rec = self.mlink.get_by_id(post_id)
if not rec:
kwd = {
'info': '您要找的分类不存在。',
}
self.render('html/404.html', kwd=kwd)
return False
kwd = {
'pager': '',
'editable': self.editable(),
}
self.render('{0}/{1}/link_view.html'.format(self.tmpl_name, self.tmpl_router),
view=rec,
unescape=tornado.escape.xhtml_unescape,
kwd=kwd,
userinfo=self.userinfo,
cfg=config.cfg,
)
@tornado.web.authenticated
def p_user_add_link(self):
if self.userinfo.privilege[1] == '1':
pass
else:
return False
post_data = {}
for key in self.request.arguments:
post_data[key] = self.get_arguments(key)
post_data['user_name'] = self.get_current_user()
cur_uid = tools.get_uudd(2)
while self.mlink.get_by_id(cur_uid):
cur_uid = tools.get_uudd(2)
if self.mlink.insert_data(cur_uid, post_data):
output = {
'addinfo ': 1,
}
else:
output = {
'addinfo ': 0,
}
return json.dump(output, self)
# self.redirect('/link/list'.format(cur_uid))
@tornado.web.authenticated
def user_add_link(self):
if self.userinfo.privilege[1] == '1':
pass
else:
return False
post_data = {}
for key in self.request.arguments:
post_data[key] = self.get_arguments(key)
post_data['user_name'] = self.get_current_user()
cur_uid = tools.get_uudd(2)
while self.mlink.get_by_id(cur_uid):
cur_uid = tools.get_uudd(2)
uid = self.mlink.insert_data(cur_uid, post_data)
self.redirect('/link/list'.format(cur_uid))
@tornado.web.authenticated
def delete(self, del_id):
if self.tmpl_router == "link":
is_deleted = self.mlink.delete(del_id)
if is_deleted:
self.redirect('/link/list')
else:
return False
else:
if self.mlink.delete(del_id):
output = {
'del_link': 1
}
else:
output = {
'del_link': 0,
}
return json.dump(output, self)
def setup(self):
print('setup 方法执行于本类中每条用例之前')
self.uu = MLink()
self.id = tools.get_uu4d()
self.raw_count = self.uu.get_counts()
print(self.raw_count)
class TestUM():
def setup(self):
print('setup 方法执行于本类中每条用例之前')
self.uu = MLink()
self.id = tools.get_uu4d()
self.raw_count = self.uu.get_counts()
print(self.raw_count)
def test_insert(self):
uid = self.id
post_data = {
'name': ['asdf'],
'link': ['sadf'],
'order': ['1'],
'logo': ['asf'],
}
self.uu.insert_data(uid, post_data)
new_count = self.uu.get_counts()
tt = self.uu.get_by_uid(uid)
# assert
assert tt.name == post_data['name'][0]
assert tt.link == post_data['link'][0]
assert tt.order == int(post_data['order'][0])
assert tt.logo == post_data['logo'][0]
assert self.raw_count + 1 == new_count
def test_upate(self):
uid = self.id
post_data = {
'name': ['asdf'],
'link': ['sadf'],
'order': ['1'],
'logo': ['asf'],
}
self.uu.insert_data(uid, post_data)
new_count = self.uu.get_counts()
assert self.raw_count + 1 == new_count
post_data2 = {
'name': ['asdlkjf'],
'link': ['sakljdf'],
'order': ['13'],
'logo': ['asfa'],
}
self.uu.update(uid, post_data2)
new_count = self.uu.get_counts()
assert self.raw_count + 1 == new_count
tt = self.uu.get_by_uid(uid)
assert tt.name != post_data['name'][0]
assert tt.link != post_data['link'][0]
assert tt.order != int(post_data['order'][0])
assert tt.logo != post_data['logo'][0]
assert tt.name == post_data2['name'][0]
assert tt.link == post_data2['link'][0]
assert tt.order == int(post_data2['order'][0])
assert tt.logo == post_data2['logo'][0]
def tearDown(self):
print("function teardown")
self.uu.delete(self.id)
def initialize(self):
self.init()
self.mpost = MPost()
self.mcat = MCatalog()
self.mpage = MPage()
self.mlink = MLink()
def render(self, num=10):
self.mlink = MLink()
recs = self.mlink.query_link(num)
return self.render_string('doc/modules/link_list.html',
recs=recs,
)
def initialize(self):
self.init()
self.user_name = self.get_current_user()
self.tmpl_router = 'link_ajax'
self.mlink = MLink()
def initialize(self):
self.init()
self.mlink = MLink()
self.tmpl_router = 'link'
class LinkHandler(BaseHandler):
def initialize(self):
self.init()
self.mlink = MLink()
self.tmpl_router = 'link'
def get(self, url_str=''):
url_arr = self.parse_url(url_str)
if len(url_arr) == 1 and url_str.endswith('.html'):
self.wiki(url_str.split('.')[0])
elif url_str == 'add_link':
self.to_add_link()
elif url_str == 'list':
self.recent()
elif url_str == 'refresh':
self.refresh()
elif url_arr[0] == 'modify':
self.to_modify(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=''):
if url_str == '':
return
print(url_str)
url_arr = url_str.split('/')
if url_arr[0] == 'modify':
self.update(url_arr[1])
elif url_str == 'add_link':
self.p_user_add_link()
elif url_arr[0] == 'add_link':
self.p_user_add_link()
else:
self.redirect('html/404.html')
def recent(self):
kwd = {
'pager': '',
'unescape': tornado.escape.xhtml_unescape,
'title': '最近文档',
}
self.render(
'doc/{0}/link_list.html'.format(self.tmpl_router),
kwd=kwd,
view=self.mlink.query_recent(),
format_date=tools.format_date,
userinfo=self.userinfo,
)
def refresh(self):
kwd = {
'pager': '',
'title': '最近文档',
}
self.render(
'doc/link/link_list.html',
kwd=kwd,
userinfo=self.userinfo,
view=self.mlink.query_dated(10),
format_date=tools.format_date,
unescape=tornado.escape.xhtml_unescape,
)
def get_random(self):
return self.mlink.query_random()
def wiki(self, uid):
dbdate = self.mlink.get_by_id(uid)
if dbdate:
self.viewit(uid)
else:
self.to_add(uid)
def to_add_link(self, ):
if self.check_doc_priv(self.userinfo)['ADD']:
pass
else:
return False
kwd = {
'pager': '',
'uid': '',
}
self.render(
'doc/{0}/link_add.html'.format(self.tmpl_router),
topmenu='',
kwd=kwd,
userinfo=self.userinfo,
)
@tornado.web.authenticated
def to_add(self, uid):
kwd = {
'uid': uid,
'pager': '',
}
self.render(
'doc/{0}/link_add.html'.format(self.tmpl_router),
kwd=kwd,
)
def __could_edit(self, uid):
raw_data = self.mlink.get_by_id(uid)
if not raw_data:
return False
if self.check_doc_priv(
self.userinfo
)['EDIT'] or raw_data.id_user == self.userinfo.user_name:
return True
else:
return False
@tornado.web.authenticated
def update(self, uid):
if self.__could_edit(uid):
pass
else:
return False
post_data = {}
for key in self.request.arguments:
post_data[key] = self.get_arguments(key)
post_data['user_name'] = self.get_current_user()
if self.tmpl_router == "link":
self.mlink.update(uid, post_data)
self.redirect('/link/list'.format(uid))
else:
if self.mlink.update(uid, post_data):
output = {
'addinfo ': 1,
}
else:
output = {
'addinfo ': 0,
}
return json.dump(output, self)
@tornado.web.authenticated
def to_modify(self, id_rec):
# 用户具有管理权限，
# 或
# 文章是用户自己发布的。
if self.__could_edit(id_rec):
pass
else:
return False
a = self.mlink.get_by_id(id_rec)
kwd = {
'pager': '',
}
self.render(
'doc/{0}/link_edit.html'.format(self.tmpl_router),
kwd=kwd,
unescape=tornado.escape.xhtml_unescape,
dbrec=a,
userinfo=self.userinfo,
)
@tornado.web.authenticated
def viewit(self, post_id):
rec = self.mlink.get_by_id(post_id)
if not rec:
kwd = {
'info': '您要找的分类不存在。',
}
self.render('html/404.html', kwd=kwd)
return False
kwd = {
'pager': '',
'editable': self.editable(),
}
self.render(
'doc/{0}/link_view.html'.format(self.tmpl_router),
view=rec,
unescape=tornado.escape.xhtml_unescape,
kwd=kwd,
userinfo=self.userinfo,
cfg=config.cfg,
)
@tornado.web.authenticated
def p_user_add_link(self):
if self.check_doc_priv(self.userinfo)['ADD']:
pass
else:
return False
post_data = {}
for key in self.request.arguments:
post_data[key] = self.get_arguments(key)
post_data['user_name'] = self.get_current_user()
cur_uid = tools.get_uudd(2)
while self.mlink.get_by_id(cur_uid):
cur_uid = tools.get_uudd(2)
if self.mlink.insert_data(cur_uid, post_data):
output = {
'addinfo ': 1,
}
else:
output = {
'addinfo ': 0,
}
return json.dump(output, self)
# self.redirect('/link/list'.format(cur_uid))
@tornado.web.authenticated
def user_add_link(self):
if self.check_doc_priv(self.userinfo)['ADD']:
pass
else:
return False
post_data = {}
for key in self.request.arguments:
post_data[key] = self.get_arguments(key)
post_data['user_name'] = self.get_current_user()
cur_uid = tools.get_uudd(2)
while self.mlink.get_by_id(cur_uid):
cur_uid = tools.get_uudd(2)
uid = self.mlink.insert_data(cur_uid, post_data)
self.redirect('/link/list'.format(cur_uid))
@tornado.web.authenticated
def delete(self, del_id):
if self.check_doc_priv(self.userinfo)['DELETE']:
pass
else:
return False
if self.tmpl_router == "link":
is_deleted = self.mlink.delete(del_id)
if is_deleted:
self.redirect('/link/list')
else:
return False
else:
if self.mlink.delete(del_id):
output = {'del_link': 1}
else:
output = {
'del_link': 0,
}
return json.dump(output, self)
def initialize(self):
self.init()
self.mpost = MPost()
self.mcat = MCatalog()
self.mpage = MPage()
self.mlink = MLink()