Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
AndrewVSutherland
GitHub Repository: AndrewVSutherland/lmfdb
Path: blob/main/scripts/belyi/old/update_db.py
1128 views
1
2
import db
3
import os, sys, inspect
4
5
Cw = db.getDBconnection_write();
6
7
belyidb = Cw['belyi']
8
belyidb.drop_collection('passports_new');
9
belyidb.drop_collection('galmaps_new');
10
passports = belyidb['passports_new']
11
galmaps = belyidb['galmaps_new']
12
13
# import the raw data
14
# import ola
15
from raw_data import ola
16
# insert passports one at a time
17
passports_upload = [ elt[0] for elt in ola];
18
galmaps_upload = [];
19
for elt in ola:
20
galmaps_upload += elt[1];
21
22
passports.insert_many(passports_upload);
23
galmaps.insert_many(galmaps_upload);
24
print(passports.find().count(), len(passports_upload))
25
print(galmaps.find().count(), len(galmaps_upload))
26
27
28
filename = inspect.getframeinfo(inspect.currentframe())[0];
29
folder = os.path.dirname(os.path.abspath(filename));
30
sys.path.append(os.path.join(folder, "../../"));
31
# from data_mgt.utilities.rewrite import create_random_object_index, rewrite_collection
32
from data_mgt.utilities.rewrite import create_random_object_index
33
if "galmaps" in belyidb.collection_names():
34
belyidb["galmaps"].rename("galmaps_old")
35
galmaps.rename("galmaps");
36
if "passports" in belyidb.collection_names():
37
belyidb["passports"].rename("passports_old")
38
passports.rename("passports");
39
create_random_object_index(belyidb, "galmaps");
40
belyidb.drop_collection('passports_old');
41
belyidb.drop_collection('galmaps_old');
42
43