Skip to content

Commit 5c94fa3

Browse files
committed
fix data_dir in util
1 parent 1b0b00a commit 5c94fa3

File tree

3 files changed

+7
-11
lines changed

3 files changed

+7
-11
lines changed

app.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def editor():
177177
if headers == []:
178178
error_msg = "Can't parse the source file "
179179
return render_template('msg.html', msg=error_msg, msg_title="Error")
180-
labels = util.get_classes_as_txt(ontologies)
180+
labels = util.get_classes_as_txt(ontologies, data_dir=DATA_DIR)
181181
# f = open(os.path.join(DATA_DIR, "labels.txt"))
182182
return render_template('editor.html', labels_txt=labels, ontologies_txt=",".join(ontologies), headers=headers,
183183
callback=callback_url, file_name=fname, error_msg=error_msg, warning_msg=warning_msg)
@@ -187,7 +187,7 @@ def editor():
187187
def get_properties():
188188
ontologies_txt = request.args.get('ontologies')
189189
ontologies = ontologies_txt.split(',')
190-
return jsonify({'properties': util.get_properties_as_list(ontologies)})
190+
return jsonify({'properties': util.get_properties_as_list(ontologies, data_dir=DATA_DIR)})
191191

192192

193193
@app.route("/get_properties_autocomplete")

generate_lookup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def build_property_lookup(dataset_name,properties_fdir, data_dir):
9292
:return:
9393
"""
9494
print("build_property_lookup> dataset_name: "+dataset_name)
95-
properties = util.get_properties_as_list([dataset_name])
95+
properties = util.get_properties_as_list([dataset_name], data_dir=data_dir)
9696
# start_idx = predict_base_URL(properties)
9797
lookup_name = 'lookup'
9898
lookup_folder_dir = os.path.join(data_dir,dataset_name,lookup_name)

util.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,6 @@
33
import os
44

55

6-
#DATA_DIR = os.path.join(BASE_DIR, 'data')
7-
DATA_DIR = 'data'
8-
9-
106
def get_random_string(length=4):
117
return ''.join(random.choice(string.lowercase) for i in range(length))
128

@@ -218,26 +214,26 @@ def get_classes_from_file(odir):
218214
return classes
219215

220216

221-
def get_properties_as_list(ontologies):
217+
def get_properties_as_list(ontologies, data_dir):
222218
"""
223219
:param ontologies:
224220
:return:
225221
"""
226222
properties = []
227223
for o in ontologies:
228-
odir = os.path.join(DATA_DIR, o, 'properties.txt')
224+
odir = os.path.join(data_dir, o, 'properties.txt')
229225
properties += get_classes_from_file(odir)
230226
return properties
231227

232228

233-
def get_classes_as_txt(ontologies):
229+
def get_classes_as_txt(ontologies, data_dir):
234230
"""
235231
:param ontologies:
236232
:return:
237233
"""
238234
classes = []
239235
for o in ontologies:
240-
odir = os.path.join(DATA_DIR, o, 'classes.txt')
236+
odir = os.path.join(data_dir, o, 'classes.txt')
241237
classes += get_classes_from_file(odir)
242238
# return classes
243239
txt = ""

0 commit comments

Comments
 (0)