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
|
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from model import *
path = "root:helloworld@localhost/piztor"
class UserData:
def __init__(self, username, password, comp_id, sec_id, sex):
self.username = username
self.password = password
self.comp_id = comp_id
self.sec_id = sec_id
self.sex = sex
def create_database():
engine = create_engine('mysql://' + path, echo = True)
Base.metadata.drop_all(engine)
Base.metadata.create_all(engine)
def import_user_data(data):
engine = create_engine('mysql://' + path, echo = True)
Session = sessionmaker(bind = engine)
session = Session()
for user in data:
um = UserModel(username = user.username,
comp_id = user.comp_id,
sec_id = user.sec_id,
sex = user.sex)
um.auth = UserAuth(user.password)
um.location = LocationInfo(lat = 0, lng = 0)
session.add(um)
session.commit()
if __name__ == '__main__':
from sys import argv, exit
if len(argv) != 2:
print "Usage: " + argv[0] + " FILE"
exit(0)
data = list()
with open(argv[1], 'r') as f:
while True:
line = f.readline().split()
if len(line) == 0: break
data.append(UserData(username = line[0],
password = line[1],
comp_id = line[2],
sec_id = line[3],
sex = line[4]))
create_database()
import_user_data(data)
|