class SimpleWidgetTestCase(unittest.TestCase): def setUp(self): # from model.info_model import MInfo self.payload = {'user_name': 'value1', 'user_pass': '******', 'user_pass2': 'fdfsrwe', 'xemail': '*****@*****.**', 'xtel': '131321342'} self.muser = MUser('value1') self.r = requests.post('http://192.168.4.166:8004/member/regist', self.payload) self.r2 = requests.post('http://192.168.4.166:8004/member/regist', self.payload) def tearDown(self): self.muser.delete_user()
def setUp(self): # from model.info_model import MInfo self.payload = {'user_name': 'value1', 'user_pass': '******', 'user_pass2': 'fdfsrwe', 'xemail': '*****@*****.**', 'xtel': '131321342'} self.muser = MUser('value1') self.r = requests.post('http://192.168.4.166:8004/member/regist', self.payload) self.r2 = requests.post('http://192.168.4.166:8004/member/regist', self.payload)
def initialize(self, hinfo=''): self.init_condition() self.mlink = MLink() self.mcat = MCatalog() self.muser_info = MUser(self.user_name) self.mcity = MCity() vip_cat = self.muser_info.get_by_username().vip_cat.strip().split(',') if vip_cat == ['']: self.vip_cat = [] else: self.vip_cat = vip_cat
class LinkHandler(base_handler.BaseHandler): def initialize(self, hinfo=''): self.init_condition() self.mlink = MLink() self.mcat = MCatalog() self.muser_info = MUser(self.user_name) self.mcity = MCity() vip_cat = self.muser_info.get_by_username().vip_cat.strip().split(',') if vip_cat == ['']: self.vip_cat = [] else: self.vip_cat = vip_cat def get(self, input=''): # Todo: 验证 if input == 'add': self.to_add() elif input == 'list': self.list() elif input.startswith('delete'): self.delete_by_id(input) elif input.startswith('edit'): self.to_edit(input) else: op, catid = input.split(r'/') if op == 'get_list': self.get_list() @tornado.web.authenticated def delete_by_id(self, input): op, uid = input.split('/') self.mlink.delete_by_uid(uid) self.list() def post(self, input=''): if input == 'add': self.add() elif input.startswith('edit'): self.update() @tornado.web.authenticated def add(self): post_data = {} for key in self.request.arguments: post_data[key] = self.get_arguments(key) file_dict = self.request.files['link_image'][0] # 对上传的图片进行处理 file_up_str = libs.upload.upload_imgfile(file_dict) out_arr = {'img': file_up_str} out_arr['url'] = post_data['link_url'][0] out_arr['catid'] = post_data['catid'][0] out_arr['cityid'] = self.city_name out_arr['parentid'] = post_data['catid'][0][:2] + '00' out_arr['username'] = self.user_name self.mlink.insert_rec(out_arr) self.redirect('/link/list') def update(self): post_data = {} for key in self.request.arguments: post_data[key] = self.get_arguments(key) if 'elink_image' in self.request.files: file_dict = self.request.files['elink_image'][0] file_up_str = libs.upload.upload_imgfile(file_dict) out_arr = {'img': file_up_str} else: out_arr = {'img': post_data['old_img'][0]} out_arr['url'] = post_data['elink_url'][0] out_arr['catid'] = post_data['ecatid'][0] self.mlink.update(post_data['uid'][0], out_arr) self.redirect('/link/list') def list(self): cat_str = '''<option value='' selected >请选择</option>''' for cat_id in self.vip_cat: cat_name = self.mcat.get_by_id(cat_id).catname tmp_str = '''<option value="%s" >%s</option>''' % (cat_id, cat_name) cat_str += tmp_str infos = self.mlink.get_links_by_cityid(self.user_name,self.city_name) kwd = { 'userid': self.user_name, 'cityname': '', 'cityid': self.city_name, 'cat_str': cat_str, 'parentid': '0000', 'parentlist': self.mcat.get_parent_list(), } self.render('link/list.html', infos=infos, kwd=kwd, userinfo=self.muser_info.get_by_username()) def to_add(self): cat_str = '''<option value='' selected >请选择</option>''' for cat_id in self.vip_cat: cat_name = self.mcat.get_by_id(cat_id).catname tmp_str = '''<option value="%s" >%s</option>''' % (cat_id, cat_name) cat_str += tmp_str kwd = { 'userid': self.user_name, 'cityid': self.city_name, 'cityname': self.mcity.get_cityname_by_id(self.city_name), 'cat_str': cat_str, 'parentid': '0000', 'parentlist': self.mcat.get_parent_list(), } self.render('link/list.html', kwd=kwd) def to_edit(self, input): op, uid = input.split('/') linkinfo = self.mlink.get_by_id(uid) tmp_s1 = '' tmp_s2 = '' tmp_s3 = '' link_cityid = linkinfo.cityid.strip() if link_cityid == 'changchun': tmp_s1 = 'selected' elif link_cityid == 'tonghua': tmp_s2 = 'selected' elif link_cityid == 'jilin': tmp_s3 = 'selected' cat_str = '''<option value='' selected >请选择</option> ''' for cat_id in self.vip_cat: cat_name = self.mcat.get_by_id(cat_id).catname tmp_str_jcc = '' if cat_id == linkinfo.catid: tmp_str_jcc = 'selected=selected' tmp_str = '''<option value="{0}" {1}>{2}</option>'''.format(cat_id, tmp_str_jcc, cat_name) cat_str += tmp_str kwd = { 'userid': self.user_name, 'cityid': self.city_name, 'cityname': self.mcity.get_cityname_by_id(self.city_name), 'cat_str': cat_str, 'parentid': '0000', 'parentlist': self.mcat.get_parent_list(), } self.render('link/list.html', kwd=kwd, linkinfo=linkinfo)