PsL4U3WZ

1---- CLASS DbConnection.py -----
2
3import psycopg2
4
5
6class DbConnection:
7
8
9    def __init__(self,dbname, user, host, pw):
10        try:
11            self.connection = psycopg2.connect(
12                "dbname="+dbname+" user ="+user+" host="+host+" password="+pw)
13            self.connection.autocommit= True;
14        except:
15            print("Cannot connect to database")
16
17----------- END ----------
18
19
20
21-------- CLçASS employee.py ----
22
23class Employee:
24
25    def __init__(self,name,surname,pay):
26        self.name = name
27        self.surname=surname
28        self.email = name + surname + '@gmail.com'
29        self.pay = pay
30
31    def __repr__(self):
32        return repr(self.name)+", "+repr(self.surname)+", "+repr(self.email)+", "+repr(self.pay )
33
34----- END ----
35
36
37----CLASS EmployeeDAO.py  ----
38
39from DbConnection import DbConnection
40from employee import Employee
41from pprint import pprint
42
43
44class EmployeeDAO:
45
46    dbConn = None
47
48    def __init__(selfs):
49        EmployeeDAO.dbConn = DbConnection ('dbname', 'user','host','pw')
50
51
52    def selectAll(self):
53        em = Employee('Yuri', 'Attanasio', 2000)
54        em1 = Employee('Danilo', 'Ciano', 2000)
55        em2 = Employee('Pino', 'Cicala', 2000)
56        emlist = {em,em1,em2}
57        pprint(emlist)
58        return emlist
59        #select * from db
60
61    def delete(self,em):
62        pprint(em);
63
64    def insert(self,em):
65        pass
66
67    def query (self, query):
68        pass
69-------- END ------------------
70
71
72---------- main -------
73
74from employee import Employee
75from EmployeeDAO import EmployeeDAO
76
77
78class main:
79
80
81
82
83    # em.print()
84    # em1.print()
85    # em2.print()
86
87    em = Employee('Yuri', 'Attanasio', 2000)
88    EmpDao = EmployeeDAO()
89    #EmpDao.delete(em)
90    #EmpDao.selectAll()