def get_req_description(self, colored=True):
requirements = []
for pub_type, required in self.requirement.items():
if colored:
requirement = oat.colorize(required["text"] + " for " + pub_type, required["color"])
else:
requirement = required["text"] + " for " + pub_type
requirements.append(requirement)
return ", ".join(requirements)
def _prepare_ezb_info(issn):
info = EZBS.get_ezb_info(issn)
if not info["success"]:
msg = "\n\nNo Information could be obtained from the EZB for ISSN " + issn
return msg
msg = "\n\nThe following additional information could be obtained from the EZB for ISSN " + issn + ":\n"
for record in info["data"]:
msg += "Title: " + record["title"] + "\n"
msg += "Access: " + oat.colorize(record["access_msg"],
record["access_color"]) + "\n"
msg += "Comments: " + str(record["remarks"]) + "\n"
msg += "-------------------------------------------------------\n"
return msg
def main():
parser = argparse.ArgumentParser()
parser.add_argument("csv_file", help=ARG_HELP_STRINGS["csv_file"])
args = parser.parse_args()
for file_path in [APC_DE_FILE, TA_FILE]:
with open(file_path, "r") as path:
reader = csv.DictReader(path)
oat.print_b("Preparing mapping tables from " + file_path + "...")
for line in reader:
data = {
"journal_full_title": line["journal_full_title"],
"publisher": line["publisher"],
"is_hybrid": line["is_hybrid"],
"count": 1
}
for issn_type in ISSN_DICTS.keys():
issn = line[issn_type]
if issn not in ISSN_DICTS[issn_type]:
ISSN_DICTS[issn_type][issn] = data
else:
ISSN_DICTS[issn_type][issn]["count"] += 1
if reader.line_num % 10000 == 0:
oat.print_b(str(reader.line_num) + " lines processed")
modified_content = []
header = None
with open(args.csv_file) as csv_file:
reader = csv.DictReader(csv_file)
header = list(reader.fieldnames)
stopped = False
for line in reader:
if stopped:
modified_content.append(line)
continue
for issn_type in ISSN_DICTS.keys():
issn = line[issn_type]
if not oat.has_value(issn):
continue
if issn in ISSN_DICTS[issn_type]:
for field_type in [
"is_hybrid", "publisher", "journal_full_title"
]:
new_value = line[field_type]
established_value = ISSN_DICTS[issn_type][issn][
field_type]
if new_value != established_value and not is_whitelisted(
field_type, new_value, established_value,
line["issn"], line["issn_print"],
line["issn_electronic"], line["issn_l"]):
msg = MISMATCH_MSG.format(
reader.line_num,
oat.colorize(field_type, "cyan"), issn_type,
issn, line["is_hybrid"], line["publisher"],
line["journal_full_title"],
oat.colorize(
str(ISSN_DICTS[issn_type][issn]["count"]),
"cyan"),
ISSN_DICTS[issn_type][issn]["is_hybrid"],
ISSN_DICTS[issn_type][issn]["publisher"],
ISSN_DICTS[issn_type][issn]
["journal_full_title"])
print(msg)
ask_msg = CORRECT_MSG.format(
field_type,
oat.colorize(established_value, "green"),
field_type,
oat.colorize(established_value, "green"))
ezb_msg = None
ret = input(ask_msg)
while ret not in ["1", "2", "3", "4"]:
if ret == "5":
if ezb_msg is None:
ezb_msg = _prepare_ezb_info(issn)
print(ezb_msg)
ret = input(
"Please select an option from 1 to 5 > ")
print("\n\n\n\n")
if ret in ["1", "2"]:
line[field_type] = established_value
if ret in ["2", "4"]:
stopped = True
break
modified_content.append(line)
modified_lines = [header]
for line in modified_content:
modified_lines.append(list(line.values()))
with open('out.csv', 'w') as out:
writer = oat.OpenAPCUnicodeWriter(out, QUOTE_MASK, True, True, False)
writer.write_rows(modified_lines)