def refresh_expense(expenseid): sql = "select userid, notes, date, createtime from expenses where expenseid=%s;" % (expenseid) out = Expense.query_sql(sql) if len(out) == 1: userid = out[0][0] content = out[0][1] date = out[0][2] createtime = out[0][3] if Expense.delete_obj_by_id(userid, expenseid) is True: ret_content = add_expense(userid, content) conn = MySQLdb.connect(config.DB_HOST, config.DB_USERNAME, config.DB_PWD, \ config.DB_NAME, charset='utf8mb4', port=config.DB_PORT) cursor = conn.cursor() sql_format = "update expenses as a inner join (select expenseid from " + \ " expenses where userid='%s' and notes='%s' limit 1) " + \ "as b on a.expenseid = b.expenseid set a.date='%s';" sql = sql_format % (userid, content, date) cursor.execute(sql) conn.commit() sql_format = "update expenses as a inner join (select expenseid from " + \ " expenses where userid='%s' and notes='%s' limit 1) " + \ "as b on a.expenseid = b.expenseid set a.createtime='%s';" sql = sql_format % (userid, content, createtime) cursor.execute(sql) conn.commit() print content, ret_content
def refresh_expense(expenseid): sql = "select userid, notes, date, createtime from expenses where expenseid=%s;" % ( expenseid) out = Expense.query_sql(sql) if len(out) == 1: userid = out[0][0] content = out[0][1] date = out[0][2] createtime = out[0][3] if Expense.delete_obj_by_id(userid, expenseid) is True: ret_content = add_expense(userid, content) conn = MySQLdb.connect(config.DB_HOST, config.DB_USERNAME, config.DB_PWD, \ config.DB_NAME, charset='utf8mb4', port=config.DB_PORT) cursor = conn.cursor() sql_format = "update expenses as a inner join (select expenseid from " + \ " expenses where userid='%s' and notes='%s' limit 1) " + \ "as b on a.expenseid = b.expenseid set a.date='%s';" sql = sql_format % (userid, content, date) cursor.execute(sql) conn.commit() sql_format = "update expenses as a inner join (select expenseid from " + \ " expenses where userid='%s' and notes='%s' limit 1) " + \ "as b on a.expenseid = b.expenseid set a.createtime='%s';" sql = sql_format % (userid, content, createtime) cursor.execute(sql) conn.commit() print content, ret_content
def delete_last_record(userid): sql_format = "select expenseid, notes from expenses" + \ " where userid='%s' order by expenseid desc limit 1" sql = sql_format % (userid) out = Expense.query_sql(sql) if len(out) == 0: content = "坑爹呀,你都没数据,删除啥?" else: expenseid = out[0][0] notes = out[0][1] if Expense.delete_obj_by_id(userid, str(expenseid)) is True: content = "上一条记录删除好了: %s。/:sun\n查看最近记录,输入: 搜索" % (notes) else: content = "Oops, 删除失败了。" return content.encode('utf-8', 'ignore')
def api_delete_expense(): userid = request.values["userid"] expenseid = request.values["expenseid"] if Expense.delete_obj_by_id(userid, expenseid) is True: content = """ { "errorcode":"200", "message":"删除成功" } """ else: content = """ { "errorcode":"500", "message":"删除失败" } """ return content
def testDeleteLastRecord(self): msg = delete_last_record(self.userid) ## print msg self.assertNotEqual(msg, "") def _assert_add_expense(self, original_content, part_result_content, print_content = False): content = add_expense(self.userid, original_content) if print_content is True: print content self.assertNotEqual(content.find(part_result_content), -1) def _assert_search_record(self, search_content, part_result_content, print_content = False): content = search_record(self.userid, search_content) if print_content is True: print content self.assertNotEqual(content.find(part_result_content), -1) def suite(): suite = unittest.TestSuite() suite.addTest(WeixinTestCase("testAddExpense1")) suite.addTest(WeixinTestCase("testAddExpense2")) suite.addTest(WeixinTestCase("testSearchRecord")) suite.addTest(WeixinTestCase("testDeleteLastRecord")) return suite if __name__ == "__main__": Expense.delete_obj_by_id(UNITTEST_USERID, "") unittest.TextTestRunner().run(suite()) ## File : weixin_unittest.py
part_result_content, print_content=False): content = add_expense(self.userid, original_content) if print_content is True: print content self.assertNotEqual(content.find(part_result_content), -1) def _assert_search_record(self, search_content, part_result_content, print_content=False): content = search_record(self.userid, search_content) if print_content is True: print content self.assertNotEqual(content.find(part_result_content), -1) def suite(): suite = unittest.TestSuite() suite.addTest(WeixinTestCase("testAddExpense1")) suite.addTest(WeixinTestCase("testAddExpense2")) suite.addTest(WeixinTestCase("testSearchRecord")) suite.addTest(WeixinTestCase("testDeleteLastRecord")) return suite if __name__ == "__main__": Expense.delete_obj_by_id(UNITTEST_USERID, "") unittest.TextTestRunner().run(suite()) ## File : weixin_unittest.py