From 7264cbfb19e8e0160f04a74c4eef91211377c69d Mon Sep 17 00:00:00 2001 From: deandr Date: Thu, 11 Feb 2021 09:26:40 -0200 Subject: [PATCH] Permite passar argumento --network para indicar a rede que o artigo pertence --- updatesearch/metadata.py | 19 +++++++++++++++++-- 1 file changed, 17 insertions(+), 2 deletions(-) diff --git a/updatesearch/metadata.py b/updatesearch/metadata.py index a661d55..d5a5e8f 100755 --- a/updatesearch/metadata.py +++ b/updatesearch/metadata.py @@ -69,10 +69,11 @@ class UpdateSearch(object): """ def __init__(self, period=None, from_date=None, until_date=None, - collection=None, issn=None, delete=False, differential=False, - load_indicators=False): + collection=None, network=None, issn=None, delete=False, + differential=False, load_indicators=False): self.delete = delete self.collection = collection + self.network = network self.from_date = from_date self.until_date = until_date self.differential = differential @@ -157,6 +158,13 @@ def pipeline_to_xml(self, article): add = ET.Element('add') for xml in xmls: + if self.network: + # Add field network passed by argument + field_network = ET.Element('field') + field_network.text = self.network + field_network.set('name', 'network') + xml.find('.').append(field_network) + add.append(xml) return ET.tostring(add, encoding="utf-8", method="xml") @@ -358,6 +366,12 @@ def main(): help='use the acronym of the collection eg.: spa, scl, col.' ) + parser.add_argument( + '-w', '--network', + default=None, + help='use to mark articles with specific network. eg.: rve, org.' + ) + parser.add_argument( '-i', '--issn', default=None, @@ -394,6 +408,7 @@ def main(): from_date=args.from_date, until_date=args.until_date, collection=args.collection, + network=args.network, issn=args.issn, delete=args.delete, differential=args.differential,