Advertisement
Guest User

ARTEMIS Database

a guest
May 31st, 2012
145
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 2.41 KB | None | 0 0
  1. import sqlite3, Artemis_Kundennummerngenerator as custgen #, Artemis_Kunde as customer
  2.  
  3. class ARTEMISDB():
  4.  
  5.     def __init__(self,accountnumber,PIN,value,currency,customerinfo):
  6.         self.accountnumber = accountnumber
  7.         self.PIN = PIN
  8.         self.value = value
  9.         self.currency = currency
  10.         self.paramlist = [self.accountnumber,self.PIN,self.value,self.currency]
  11.         self.customerinfo = customerinfo
  12.    
  13.     def connectDB(self):
  14.         self.connection = sqlite3.connect("ARTEMIS_Database.db")
  15.         self.cur = self.connection.cursor()
  16.  
  17.     def createDB(self):
  18.  
  19.         self.connectDB()
  20.         customergen = custgen.CustomerGen()
  21.         self.customernumbers = customergen.SQLCustomergenerator()
  22.         self.createTable()
  23. #        customerinformations = customer.CustomerInfo("PLACEHOLDER","PLACEHOLDER")
  24. #        self.customerinfo = customerinformations.createList()
  25. #        print self.customerinfo
  26.        
  27.     def checkDB(self):
  28.        
  29.         try:
  30.             open("ARTEMIS_Database.db")
  31.             print "Database found, loading tables"
  32.             self.loadDB()
  33.        
  34.         except IOError:
  35.             print "Database not found, creating one..."
  36.             self.createDB()
  37.    
  38.     def createTable(self):
  39.    
  40.         self.cur.execute("""CREATE TABLE Customers (
  41.        Name TEXT, Surname TEXT, Age TEXT, Customernumber TEXT)""")
  42.        
  43.         self.connection.commit()
  44.                                              
  45.         self.cur.execute("""CREATE TABLE Accounts (
  46.        Accountnumber TEXT, PIN INTEGER, Value INTEGER, Currency TEXT)""")
  47.                                              
  48.         self.connection.commit()
  49.  
  50.     def loadDB(self):
  51.         self.connectDB()
  52.        
  53.     def saveDB(self):
  54.         pass
  55.  
  56.     def editDB(self):
  57.          
  58.         self.cur.execute("INSERT INTO Accounts VALUES (?,?,?,?)",self.paramlist)                                                
  59.         self.connection.commit()
  60.        
  61.         self.customerinfo.append(self.customernumbers[0])
  62.         self.customernumbers.remove(self.customerinfo[3])
  63.        
  64.         self.cur.execute("INSERT INTO Customers VALUES (?,?,?,?)",self.customerinfo)
  65.         self.connection.commit()
  66.                        
  67. if __name__=="__main__":
  68.    
  69.     list1 = ["Frost","Byte","17"]
  70.     main = ARTEMISDB("DE001",1337,666,"EUR",list1)
  71.     main.checkDB()
  72.     main.editDB()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement