in scripts/anonymize.py [0:0]
def __init__(self, dir_path):
self.dir_path = dir_path
self.abFP = {}
if os.path.exists(os.path.join(self.dir_path, 'abFP.csv')):
print("loading abFP")
df = pd.read_csv(os.path.join(dir_path, 'abFP.csv'))
for _, row in df.iterrows():
self.abFP[row['abstractFingerPrint']] = row['id']
self.db = {}
if os.path.exists(os.path.join(self.dir_path, 'db.csv')):
print("loading db")
df = pd.read_csv(os.path.join(dir_path, 'db.csv'))
for _, row in df.iterrows():
self.db[row['db_name']] = row['id']
self.table = {}
if os.path.exists(os.path.join(self.dir_path, 'table.csv')):
print("loading table")
df = pd.read_csv(os.path.join(dir_path, 'table.csv'))
for _, row in df.iterrows():
self.table[row['table_name']] = row['id']