generic/structure_assignment
16
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

IssueID #1835: fixed use of tempfile

Browse Source
This commit is contained in:
Cyprian Laskowski 2021-03-18 09:10:27 +01:00
parent cb6960f58a
commit 59dd78f9b2
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
scripts/process.py
View File

@ -1,24 +1,26 @@
import argparse import argparse
import tempfile import tempfile
import os import shutil
from structure_assignment.pipeline import Pipeline, create_nlp from structure_assignment.pipeline import Pipeline, create_nlp
resource_directory = '../resources' resource_directory = '../resources'
def run_all(input_file_name, output_file_name, nlp, structure_file_name): def run_all(input_file_name, output_file_name, nlp, structure_file_name):
tmp_file_name = tempfile.mksfile() tmp_directory = tempfile.mkdtemp()
string_to_parse(input_file_name, tmp_file_name, nlp) tmp_file_name = tmp_directory + '/parsed.xml'
strings_to_parse(input_file_name, tmp_file_name, nlp)
parse_to_dictionary(tmp_file_name, output_file_name, structure_file_name) parse_to_dictionary(tmp_file_name, output_file_name, structure_file_name)
os.remove(tmp_file_name) shutil.rmtree(tmp_directory)
validate_structures(structure_file_name) validate_structures(structure_file_name)
validate_dictionary(output_file_name) validate_dictionary(output_file_name)
def strings_to_dictionary(input_file_name, output_file_name, nlp, structure_file_name): def strings_to_dictionary(input_file_name, output_file_name, nlp, structure_file_name):
tmp_file_name = tempfile.mksfile() tmp_directory = tempfile.mkdtemp()
string_to_parse(input_file_name, tmp_file_name, nlp) tmp_file_name = tmp_directory + '/parsed.xml'
strings_to_parse(input_file_name, tmp_file_name, nlp)
parse_to_dictionary(tmp_file_name, output_file_name, structure_file_name) parse_to_dictionary(tmp_file_name, output_file_name, structure_file_name)
os.remove(tmp_file_name) shutil.rmtree(tmp_directory)
def strings_to_parse(input_file_name, output_file_name, nlp): def strings_to_parse(input_file_name, output_file_name, nlp):
pipeline = Pipeline(resource_directory, nlp) pipeline = Pipeline(resource_directory, nlp)