forked from itucsdb1515/itucsdb1515
-
Notifications
You must be signed in to change notification settings - Fork 2
/
users.py
208 lines (176 loc) · 7.56 KB
/
users.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
import pymysql
from dbconnection import MySQL
class Users:
def __init__(self):
self.users = {}
self.key = 0
def add_user(self, user):
self.key += 1
# user.key = self.key
self.users[self.key] = user
def get_user(self, key):
return self.users[key]
def get_users(self):
return sorted(self.users.items())
class User:
def __init__(self, user_id="", user_type="", user_email="", user_password="", user_name="", user_surname="",
user_phone="", user_address=""):
self.user_id = user_id
self.user_type = user_type
self.user_email = user_email
self.user_password = user_password
self.user_name = user_name
self.user_surname = user_surname
self.user_phone = user_phone
self.user_address = user_address
if user_type == 1:
self.add_user_detail(user_name, user_surname, user_phone, user_address)
elif user_type == 2:
self.add_company_detail(user_name, user_phone, user_address)
elif user_type == 3:
self.add_university_detail(user_name, user_address)
def add_user_detail(self, user_name="", user_surname="", user_phone="", user_address=""):
self.user_name = user_name
self.user_surname = user_surname
self.user_phone = user_phone
self.user_address = user_address
print('add user detail')
def add_company_detail(self, user_name="", user_phone="", user_address=""):
self.user_name = user_name
self.user_phone = user_phone
self.user_address = user_address
print('add company detail')
def add_university_detail(self, user_name="", user_address=""):
self.user_name = user_name
self.user_address = user_address
print('add university detail')
def user_show(user_id):
user = User()
try:
conn = pymysql.connect(host=MySQL.HOST, port=MySQL.PORT, user=MySQL.USER,
passwd=MySQL.PASSWORD, db=MySQL.DB, charset=MySQL.CHARSET)
c = conn.cursor()
sql = """SELECT user_id, user_type, user_email, user_password FROM users WHERE user_id = %d """ % (
int(user_id))
c.execute(sql)
print(user_id)
for row in c:
user_id, user_type, user_email, user_password = row
user = User(user_id=user_id, user_type=user_type, user_email=user_email, user_password=user_password)
if user_type == 1:
sql = """SELECT user_name, user_surname, phone, address
FROM user_detail WHERE user_id = %d """ % (
int(user_id))
print(sql)
c.execute(sql)
for row_user in c:
user_name, user_surname, phone, address = row_user
user.add_user_detail(user_name, user_surname, phone, address)
elif user_type == 2:
sql = """SELECT company_name, company_phone, company_address
FROM company_detail WHERE user_id = %d """ % (
int(user_id))
print(sql)
c.execute(sql)
for row_user in c:
company_name, company_phone, company_address = row_user
user.add_company_detail(user_name=company_name, user_phone=company_phone, user_address=company_address)
elif user_type == 3:
sql = """SELECT university_name, university_address
FROM university_detail WHERE user_id = %d """ % (
int(user_id))
print(sql)
c.execute(sql)
for row_user in c:
university_name, university_address = row_user
user.add_user_detail(user_name=university_name, user_address=university_address)
c.close()
conn.close()
except Exception as e:
print(str(e))
print(user)
return user
def user_edit(user_id, user_name="", user_surname="", user_phone="", user_address=""):
try:
conn = pymysql.connect(host=MySQL.HOST, port=MySQL.PORT, user=MySQL.USER,
passwd=MySQL.PASSWORD, db=MySQL.DB, charset=MySQL.CHARSET)
c = conn.cursor()
sql = """SELECT user_type FROM users WHERE user_id = %d """ % (
int(user_id))
c.execute(sql)
for row in c:
user_type = row[0]
if user_type == 1:
f = '%Y-%m-%d'
print('update user detail')
sql = """UPDATE user_detail SET user_name = '%s', user_surname = '%s', phone = '%s', address = '%s' WHERE user_id = %d """ % (
user_name, user_surname, user_phone, user_address, int(user_id))
c.execute(sql)
elif user_type == 2:
print('update company detail')
sql = """UPDATE company_detail SET company_name = '%s', company_phone = '%s', company_address = '%s' WHERE user_id = %d """ % (
user_name, user_phone, user_address, int(user_id))
c.execute(sql)
elif user_type == 3:
print('update university detail')
sql = """UPDATE university_detail SET university_name = '%s', university_address = '%s' WHERE user_id = %d """ % (
user_name, user_address, int(user_id))
print(sql)
c.execute(sql)
conn.commit()
c.close()
conn.close()
except Exception as e:
print(str(e))
def user_delete(user_id):
try:
conn = pymysql.connect(host=MySQL.HOST, port=MySQL.PORT, user=MySQL.USER,
passwd=MySQL.PASSWORD, db=MySQL.DB, charset=MySQL.CHARSET)
c = conn.cursor()
sql = """DELETE FROM recommended WHERE following_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
sql = """DELETE FROM connections_detail WHERE user_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
sql = """DELETE FROM connections WHERE following_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
sql = """DELETE FROM conversations WHERE participant_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
sql = """DELETE FROM jobs WHERE user_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
sql = """DELETE FROM comment WHERE user_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
sql = """DELETE FROM posts WHERE user_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
sql = """DELETE FROM likes WHERE user_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
sql = """DELETE FROM job_appliers WHERE user_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
sql = """DELETE FROM location WHERE user_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
sql = """DELETE FROM user_detail WHERE user_id = (%d) """ % (int(user_id))
print(sql)
c.execute(sql)
sql = """DELETE FROM company_detail WHERE user_id = (%d) """ % (int(user_id))
print(sql)
c.execute(sql)
sql = """DELETE FROM university_detail WHERE user_id = (%d) """ % (int(user_id))
print(sql)
c.execute(sql)
sql = """DELETE FROM users WHERE user_id = (%d) """ % int(user_id)
print(sql)
c.execute(sql)
conn.commit()
c.close()
conn.close()
except Exception as e:
print(str(e))