Skip to content

Commit 96d3d95

Browse files
committed
Added generic loader
1 parent a275135 commit 96d3d95

File tree

3 files changed

+45
-1
lines changed

3 files changed

+45
-1
lines changed

.gitignore

+1
Original file line numberDiff line numberDiff line change
@@ -142,3 +142,4 @@ cython_debug/
142142
# Common data sources for this project
143143
US.txt
144144
facility.csv
145+
exportedDoctors.csv

load.py

+43
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
from os import getenv
2+
from dotenv import load_dotenv
3+
load_dotenv()
4+
import sys
5+
import io
6+
import logging
7+
log = logging.getLogger()
8+
9+
def setup_logger(log_level=logging.DEBUG):
10+
# setup logging to console output
11+
formatter = logging.Formatter('%(asctime)s %(name)s %(levelname)s %(message)s')
12+
handler = logging.StreamHandler()
13+
handler.setFormatter(formatter)
14+
logger = logging.getLogger()
15+
logger.setLevel(log_level)
16+
logger.addHandler(handler)
17+
return logger
18+
19+
# todo: pull from URL rather than filesystem
20+
#import requests
21+
22+
import numpy as np
23+
import pandas as pd
24+
25+
from neighbor.storage import Storage
26+
27+
import json
28+
29+
if __name__ == '__main__':
30+
if len(sys.argv) <= 1:
31+
print("A filename must be provided")
32+
quit()
33+
34+
log = setup_logger()
35+
36+
storage = Storage()
37+
storage.connect()
38+
39+
df = pd.read_csv(sys.argv[1], ',',keep_default_na=False, header=None)
40+
41+
storage.execute_batch_neighborlocation_insert(df)
42+
43+
storage.setup_neighbor_geometry()

neighbor/storage.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def execute_batch_neighborlocation_insert(self, dataframe, page_size=150):
6464
nump = dataframe.to_numpy()
6565

6666
tpls = [tuple(x) for x in nump]
67-
67+
6868
sql = """
6969
INSERT INTO Orbitron.NeighborLocations
7070
(SourceId, Name, Latitude, Longitude, RowData)

0 commit comments

Comments
 (0)