add getKeys in switchrule.py

db_transfer.py

@@ -76,7 +76,11 @@ class DbTransfer(object):
 	@staticmethod
 	def pull_db_all_user():
 		#数据库所有用户信息
-		keys = ['port', 'u', 'd', 'transfer_enable', 'passwd', 'enable' ]
+		try:
+			import switchrule
+			keys = switchrule.getKeys()
+		except Exception, e:
+			keys = ['port', 'u', 'd', 'transfer_enable', 'passwd', 'enable' ]
 		conn = cymysql.connect(host=Config.MYSQL_HOST, port=Config.MYSQL_PORT, user=Config.MYSQL_USER,
 								passwd=Config.MYSQL_PASS, db=Config.MYSQL_DB, charset='utf8')
 		cur = conn.cursor()

switchrule.py

@@ -1,3 +1,6 @@
+def getKeys():
+	return ['port', 'u', 'd', 'transfer_enable', 'passwd', 'enable' ]
+
 def isTurnOn(row):
 	return True