Das deutsche Python-Forum

Code: Alles auswählen

import csv

class Database():
    def __init__(self):
        pass

    def importDB(self, db_path):
        self.db_path = db_path
        self.database = {}
        with open(self.db_path, "r") as db:
            for key, value in csv.reader(db):
                self.database[key] = value

    def appendUser(self, userdata):
        username = "{}{}".format("Customer_", len(self.database.keys())+1)
        a, b, c, d, e, f, g, h, i, j = userdata
        self.database[username] = {'Bezeichnung':a, 'Strasse':b, 'Postleitzahl':c,
                                   'Ort':d, 'Land':e, 'Tuer':f, 'Treppe':g, 'Telefon':h,
                                   'Fax':i, 'Postfach':j}

    def delUser(self, usr):
        #search through database and delete user
        pass

    def getUser(self, usr):
        #return userdata
        return eval(self.database[usr])

    def seeker(self, key=None, retn=None):
        #search through database or/and return database
        if retn:
            return self.database
    
    def saveDB(self):
        #overwrites the old database
        with open(self.db_path, "w") as db:
            db = csv.writer(db)
            for key, val in self.database.items():
                db.writerow([key, val])



if "__main__"==__name__:
    userdata = "1", "2", "3", "4", "5", "7", "6", "8", "9", "10"
    db = Database()
    db.importDB("database.csv")
    db.appendUser(userdata)
    user = db.getUser("Customer_2")
    print user
    a = db.seeker(retn=True)
    print len(a.keys())
    db.saveDB()

Code: Alles auswählen

        column_names = [
            'Bezeichnung', 'Strasse', 'Postleitzahl', 'Ort', 'Land', 'Tuer',
            'Treppe', 'Telefon', 'Fax', 'Postfach'
        ]
        self.rows.append(dict(izip(column_names, userdata)))

Code: Alles auswählen

import csv
 
class Database(object):
    FIELDS = ()
    
    def __init__(self):
        self.database = []
 
    def importDB(self, db_path):
        with open(db_path, "r") as db:
            reader = csv.reader(db)
            header = next(reader)
            assert header, self.FIELDS
            self.database.extend(reader)
 
    def saveDB(self, db_path):
        with open(db_path, "w") as db:
            writer = csv.writer(db)
            writer.writerow(self.FIELDS)
            writer.writerows(self.database)

    def append(self, *data, **kw_data):
        data = list(data)
        print len(data),self.FIELDS[len(data):]
        for field in self.FIELDS[len(data):]:
            print field
            data.append(kw_data.pop(field))
        assert not kw_data, "Zu viele Felder angegeben: %s"%', '.join(kw_data)
        assert len(data)==len(self.FIELDS), "Zu wenige Felder angegeben"
        self.database.append(data)
 
    def get(self, index):
        return self.database[index]
    
    def query(self, **query):
        query = [(self.FIELDS.index(key),value) for key,value in query.iteritems()]
        return (entry for entry in self.database
                if all(entry[index]==value for index,value in query))
    
class UserDatabase(Database):
    FIELDS = (
        'Bezeichnung', 'Strasse', 'Postleitzahl', 'Ort', 'Land', 'Tuer',
        'Treppe', 'Telefon', 'Fax', 'Postfach'
    )
 
if "__main__"==__name__:
    userdata = "1", "2", "3", "4", "5", "7", "6", "8", "9", "10"
    db = UserDatabase()
    db.importDB('database.csv')
    db.append(*userdata)
    db.append('bla','fsd','sdfs',Ort='fds',Land='fdsf',Tuer='ofs',Treppe='',
              Telefon='06543',Fax='324',Postfach='343')
    print db.database
    print list(db.query(Ort='fds', Tuer='ofs'))
    db.saveDB('database.csv')