From e1affa2135e66c1418138750a6fcbe8dec7d7574 Mon Sep 17 00:00:00 2001 From: yamat0jp Date: Sat, 3 Dec 2016 21:53:32 +0900 Subject: [PATCH] Userdel change for test if digit --- index.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/index.py b/index.py index 189cf1c..7fad77d 100755 --- a/index.py +++ b/index.py @@ -236,12 +236,14 @@ class AdminConfHandler(BaseHandler): class UserHandler(tornado.web.RequestHandler): def post(self,dbname): - num = int(self.get_argument('number')) - pas = self.get_argument('password') - table = self.application.db[dbname] - obj = table.find_one({'number':num}) - if obj and(obj['password'] == pas): - table.remove({'number':num}) + num = self.get_argument('number') + if num.isdigit() == True: + num = int(num) + pas = self.get_argument('password') + table = self.application.db[dbname] + obj = table.find_one({'number':num}) + if obj and(obj['password'] == pas): + table.remove({'number':num}) self.redirect('/'+dbname) class SearchHandler(tornado.web.RequestHandler): -- 2.11.0