View difference between Paste ID: shJnELiv and NBpyWk6C

 #!/usr/bin/env python
 
 import MySQLdb
 import MySQLdb.cursors
 
-# Information about the database
+class MCDataSource(object):
 	# Information about the database
-database = "ENTS"
+
-table = "members"
+	database = "ENTS"
-hostaddr = "localhost"
+	table = "members"
-username = "mastercontrol"
+	hostaddr = "localhost"
-# TODO: This is insecure as it can be read by anyone with access to this script, needs to be handled in another way
+	username = "mastercontrol"
-password = "pass"
+	# TODO: This is insecure as it can be read by anyone with access to this script, needs to be handled in another way
 	password = "pass"
-def connect():
+	conn = None
-	conn = MySQLdb.connect ( host = hostaddr, user = username, passwd = password, db = database, cursorclass=MySQLdb.cursors.DictCursor )
+	cursor = None
-        cursor = conn.cursor()
+
-	return conn, cursor
+	"""Use to create a singleton"""
 	def __new__(cls, *args, **kwds):
-def close(conn, cursor):
+		"""
-	cursor.close()
+		>>> s = Singleton()
-	conn.close()
+		>>> p = Singleton()
 		>>> id(s) == id(p)
-# Go to the database and get the last unlock event for the given door number
+		True
-# We match FobNumber on the members table with CardNum on the DoorLog table and search based on door number
+		"""
-def getLastUnlock( doorNum ):
+		self = "__self__"
-        print "getLastUnlock connecting to db"
+		if not hasattr(cls, self):
-	conn, cursor = connect()
+			instance = object.__new__(cls)
-        #conn = MySQLdb.connect ( host = hostaddr, user = username, passwd = password, db = database, cursorclass=MySQLdb.cursors.DictCursor )
+			instance.init(*args, **kwds)
-        #cursor = conn.cursor()                          #connect to users db
+			setattr(cls, self, instance)
-        print "connection cursor created"
+		return getattr(cls, self)
-	# We use an inner join because we want to find at least one match on both tables
+
-        cursor.execute ("SELECT * from DoorLog inner join members ON members.FobNumber=DoorLog.CardNum where DoorNum = '"+str(doorNum)+"' order by EventDateTime DESC")
+	def init(self, *args, **kwds):
-        row = cursor.fetchone()                 #get the record for the current cardnum
+		connect()
-	print "getLastUnlock returning row:"
+
-	print row
+	def connect():
-	close(conn, cursor)
+		self.conn = MySQLdb.connect ( host = hostaddr, user = username, passwd = password, db = database, cursorclass=MySQLdb.cursors.DictCursor )
-        return row
+        	self.cursor = conn.cursor()
 		#return conn, cursor
 
-def getMember( fobNumber, fobField ):
+	def close(conn, cursor):
-        #conn = MySQLdb.connect ( host = hostaddr, user = username, passwd = password, db = database, cursorclass=MySQLdb.cursors.DictCursor )
+		self.cursor.close()
-        #cursor = conn.cursor()                          #connect to users db
+		self.conn.close()
-	print "getMember connecting to db"
+
-	conn, cursor = connect()
+	# Go to the database and get the last unlock event for the given door number
-        cursor.execute ("SELECT * from members where "+fobField+" = '" + fobNumber+"'")
+	# We match FobNumber on the members table with CardNum on the DoorLog table and search based on door number
 	def getLastUnlock( doorNum ):
-        try:
+        	print "getLastUnlock connecting to db"
-        	row = cursor.fetchone()                 #get the record for the current cardnum
+		#conn, cursor = connect()
-        except ValueError:              #Exception:
+	       	#conn = MySQLdb.connect ( host = hostaddr, user = username, passwd = password, db = database, cursorclass=MySQLdb.cursors.DictCursor )
-                #no match for that user cardNum, send
+	        #cursor = conn.cursor()                          #connect to users db
-                #server.serve_forever()back fail message
+        	print "connection cursor created"
-                print "Major Failure \n"
+		# We use an inner join because we want to find at least one match on both tables
-                #print Exception
+        	self.cursor.execute ("SELECT * from DoorLog inner join members ON members.FobNumber=DoorLog.CardNum where DoorNum = '"+str(doorNum)+"' order by EventDateTime DESC")
         	row = self.cursor.fetchone()                 #get the record for the current cardnum
-        #close the mysql connection
+		print "getLastUnlock returning row:"
-       	close(conn, cursor)
+		print row
-	return row
+		#close(conn, cursor)
         	return row
-def createDoorLogFail( fobNumber, doorNumber, cmd ):
+
-	print "createDoorLogFail connecting to db"
+
-	conn, cursor = connect()
+	def getMember( fobNumber, fobField ):
-	cursor.execute("insert into DoorLog values ("+fobNumber+
+        	#conn = MySQLdb.connect ( host = hostaddr, user = username, passwd = password, db = database, cursorclass=MySQLdb.cursors.DictCursor )
-        ",'"+fobNumber+" Not in database: "+cmd+
+        	#cursor = conn.cursor()                          #connect to users db
-        "','"+doorNumber+"',now(),NULL)")
+		print "getMember connecting to db"
-	close(conn, cursor)
+		#conn, cursor = connect()
         	self.cursor.execute ("SELECT * from members where "+fobField+" = '" + fobNumber+"'")
-def createDoorLogAdmit( fobNumber, firstName, lastName, doorNumber):
+
-	print "createDoorLogAdmit connecting to db"
+        	try:
-	conn, cursor = connect()
+        		row = self.cursor.fetchone()                 #get the record for the current cardnum
-        cursor.execute("insert into DoorLog values ("+fobNumber+
+        	except ValueError:              #Exception:
-        ",'"+firstName+" "+lastName+" unlocked door "+doorNumber+
+                	#no match for that user cardNum, send
-        "','"+doorNumber+"',now(),NULL)")
+                	#server.serve_forever()back fail message
-	close(conn, cursor)
+                	print "Major Failure \n"
                 	#print Exception
-def getSeconds( lastUnlock ):
+
-	print "getSeconds connecting to db"
+        	#close the mysql connection
-	conn, cursor = connect()
+       		#close(conn, cursor)
-	cursor.execute("select timestampdiff(second,'"+lastUnlock+"',now()) as seconds")
+		return row
-        seconds = cursor.fetchone()['seconds']
+
-	if (seconds is None):
+	def createDoorLogFail( fobNumber, doorNumber, cmd ):
-		seconds = 2000
+		print "createDoorLogFail connecting to db"
-	close(conn, cursor)
+		#conn, cursor = connect()
-	return seconds
+		self.cursor.execute("insert into DoorLog values ("+fobNumber+
         	",'"+fobNumber+" Not in database: "+cmd+
-def updateMember( fobNumber, fobField, lastUnlockField ):
+        	"','"+doorNumber+"',now(),NULL)")
-	print "updateMember connecting to db"
+		#close(conn, cursor)
-	conn, cursor = connect()
+
-	cursor.execute("update members set "+lastUnlockField+" =  now() where "+fobField+" = '"+fobNumber+"'")
+	def createDoorLogAdmit( fobNumber, firstName, lastName, doorNumber):
-	close(conn, cursor)
+		print "createDoorLogAdmit connecting to db"
 		#conn, cursor = connect()
         	self.cursor.execute("insert into DoorLog values ("+fobNumber+
         	",'"+firstName+" "+lastName+" unlocked door "+doorNumber+
         	"','"+doorNumber+"',now(),NULL)")
 		#close(conn, cursor)
 
 	def getSeconds( lastUnlock ):
 		print "getSeconds connecting to db"
 		#conn, cursor = connect()
 		self.cursor.execute("select timestampdiff(second,'"+lastUnlock+"',now()) as seconds")
         	seconds = self.cursor.fetchone()['seconds']
 		if (seconds is None):
 			seconds = 2000
 		#close(conn, cursor)
 		return seconds
 
 	def updateMember( fobNumber, fobField, lastUnlockField ):
 		print "updateMember connecting to db"
 		#conn, cursor = connect()
 		self.cursor.execute("update members set "+lastUnlockField+" =  now() where "+fobField+" = '"+fobNumber+"'")
 		#close(conn, cursor)