-
Notifications
You must be signed in to change notification settings - Fork 1
/
seed.py
102 lines (76 loc) · 3.23 KB
/
seed.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
"""Utility file to seed user database from MovieLens data in seed_data/"""
import datetime
from model import User, Position, User_Assets, connect_to_db, db
from server import app
def load_users():
"""Load users from u.user into database."""
for i, row in enumerate(open("seed_data/u.user")):
row = row.rstrip()
user_id, first_name, last_name, picture, email_address, last_login, user_LinkedIn_url, user_Twitter_url, user_Facebook_url, user_website_url = row.split("|")
user = User(user_id=user_id,
first_name=first_name,
last_name=last_name,
picture=picture,
email_address=email_address,
last_login=last_login,
user_LinkedIn_url=user_LinkedIn_url,
user_Twitter_url=user_Twitter_url,
user_Facebook_url=user_Facebook_url,
user_website_url=user_website_url)
# This adds my user to the session
db.session.add(user)
# This provide some sense of progress
if i % 100 == 0:
print i
# This commits my user to the database User
db.session.commit()
def load_positions():
"""Load position from u.position into database."""
print "Position"
for i, row in enumerate(open("seed_data/u.position")):
row = row.rstrip()
position_id, title, position_summary, deadline, company_name, location, application_status, position_url = row.split("|")
# if deadline:
# deadline_app = datetime.datetime.strptime("deadline", "%m-%d-%Y")
# else:
# deadline_app = None
# print deadline_app
print position_id
position = Position(position_id=position_id,
title=title,
position_summary=position_summary,
deadline=deadline,
company_name=company_name,
location=location,
application_status=application_status,
position_url=position_url)
# This adds my position to the session
db.session.add(position)
# This provide some sense of progress
if i % 100 == 0:
print i
# # This commits my position to the database User
db.session.commit()
def load_user_assets():
"""Load User's Assets from u.user_assets into database."""
print "User_Assets"
for i, row in enumerate(open("seed_data/u.user_assets")):
row = row.rstrip()
user_asset_id, asset_type, asset_content = row.split("|")
user_assets = User_Assets(user_asset_id=user_asset_id,
asset_type=asset_type,
asset_content=asset_content)
# print title
# This adds my position to the session
db.session.add(user_assets)
# This provide some sense of progress
if i % 100 == 0:
print i
# This commits my position to the database User
db.session.commit()
if __name__ == "__main__":
connect_to_db(app)
db.create_all()
load_users()
load_user_assets()
load_positions()