12
12
_PASSWORD = 'password'
13
13
_HOST = 'localhost'
14
14
15
- class RequestHandler (BaseHTTPRequestHandler ):
16
-
17
- def __init__ (self , request , client_address , server ):
18
- print ("psychopg init" )
19
- super ().__init__ (request , client_address , server )
20
- self .prepare_database ()
21
15
22
- def prepare_database (self ) -> None :
23
-
24
- conn = psycopg2 .connect (dbname = _DBNAME , user = _USER , password = _PASSWORD , host = _HOST )
16
+ def prepare_database () -> None :
17
+ conn = psycopg2 .connect (dbname = _DBNAME , user = _USER , password = _PASSWORD , host = _HOST )
25
18
26
- print ("db connected" )
19
+ print ("db connected" )
27
20
28
- cur = conn .cursor ()
21
+ cur = conn .cursor ()
29
22
30
- cur .execute ("""
23
+ cur .execute ("""
31
24
CREATE TEMPORARY TABLE test_table (
32
25
id SERIAL PRIMARY KEY,
33
26
name TEXT NOT NULL
34
27
)
35
28
""" )
36
29
37
- cur .execute ("INSERT INTO test_table (name) VALUES (%s)" , ("Alice" ,))
38
- cur .execute ("INSERT INTO test_table (name) VALUES (%s)" , ("Bob" ,))
30
+ cur .execute ("INSERT INTO test_table (name) VALUES (%s)" , ("Alice" ,))
31
+ cur .execute ("INSERT INTO test_table (name) VALUES (%s)" , ("Bob" ,))
39
32
40
- conn .commit ()
33
+ conn .commit ()
41
34
42
- cur .close ()
43
- conn .close ()
35
+ cur .close ()
36
+ conn .close ()
44
37
38
+
39
+ class RequestHandler (BaseHTTPRequestHandler ):
45
40
@override
46
41
# pylint: disable=invalid-name
47
42
def do_GET (self ):
@@ -66,7 +61,9 @@ def handle_request(self):
66
61
conn .close ()
67
62
self .send_response (200 , "success" )
68
63
69
- def main ()-> None :
64
+
65
+ def main () -> None :
66
+ prepare_database ()
70
67
server_address : Tuple [str , int ] = ("0.0.0.0" , _PORT )
71
68
request_handler_class : type = RequestHandler
72
69
requests_server : ThreadingHTTPServer = ThreadingHTTPServer (server_address , request_handler_class )
@@ -76,5 +73,6 @@ def main()->None:
76
73
print ("Psychopg2-Ready" )
77
74
server_thread .join ()
78
75
76
+
79
77
if __name__ == "__main__" :
80
78
main ()
0 commit comments