def insert_names_into_db(batch_size: int = 10000, limit: int = 200000):
with open('name.basics.tsv') as names_tsv:
names_tsv.readline()
reader = csv.reader(names_tsv, delimiter='\t')
counter = 0
names = []
number_in_batch = 0
for row in tqdm(reader):
if number_in_batch > batch_size:
number_in_batch = 0
db_session.add_all(names)
db_session.commit()
names = []
counter += batch_size
print(f'added {counter} names')
if limit and counter >= limit:
return
row = list(map(map_null, row))
name = Name(
nconst=row[0],
primaryName=row[1],
birthYear=row[2],
deathYear=row[3],
primaryProfession=row[4],
)
if row[5]:
titles = Title.query.filter(Title.tconst.in_(
row[5].split(',')))
name.knownForTitles = list(titles)
names.append(name)
number_in_batch += 1
db_session.commit()
def name():
if request.method == 'POST':
if 'Projects' in request.form:
return redirect(url_for('projects'))
elif 'About' in request.form:
return redirect(url_for('about'))
names = request.form['name']
n = Name(name=names)
db.session.add(n)
db.session.commit()
'''
print('lalalal')
details = request.form
namez = details['name']
namez=Names(name=namez,)
db.session.add(namez)
db.session.commit()
'''
return render_template('Names.html', name=Names.query.all())
else:
return render_template('Names.html', name=Names.query.all())
def add_name():
# add a name to the DB
new_name = Name()
new_name.id = determine_next_id()
new_name.name = request.json["name"]
new_name.gender = request.json["gender"]
db.session.add(new_name)
db.session.commit()
return jsonify(success=True)
def add_name():
name = request.values.get('name')
email = request.values.get('email')
contact_number = request.values.get('contact_number')
name = Name(name=name, email=email, contact_number=contact_number)
db.session.add(name)
db.session.commit()
return "Successfully added"
incarceration_type=models.IncarcerationType.JAIL,
duration=timedelta(days=90)),
fine=False))
ARREST_INFO_2 = models.ArrestInfo(arrest_id='9483488DIP657',
name_as_charged_id=2,
date=datetime(1994, 6, 10),
city='SAN JOSE')
EVENT_2 = models.Event(
arrest_info=ARREST_INFO_2,
listed_dob=datetime(1953, 4, 20),
associated_cases=[CASE_INFO_2_1],
probation_modifications=[] # todo
)
RAP_SHEET_1 = models.RAPSheet(names_as_charged=[
Name('ALEXANDER', 'ROBIN RAE'),
Name('MOAN', 'ROBIN', 'RAE'),
Name('LEDEL', 'ROBIN', 'RAE'),
Name('LEDEL', 'ROBIN'),
Name('SUMMERS', 'KARLA'),
Name('LEDEL', 'RUBIN', 'RAE'),
Name('SUMMERS', 'CARTA'),
Name('TYREE', 'ROBIN'),
Name('TYREE', 'ROBIN', 'RAE'),
Name('ALEXANDER', 'ROBIN'),
Name('TYREE', 'ROBIN', 'R')
],
dob=datetime(1953, 4, 20, 0, 0),
sex='F',
events=[EVENT_0, EVENT_1, EVENT_2])
def create_name(db, name, contest):
name = Name(name=name, contest=contest)
add_and_commit(db, name)
return name
from models import Name # models에 .붙이면 에러남
name = Name('Pewww')
print(name.getName())
