def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_keywords_repeater.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.ui.thesaurus_label = self.thesaurus_label
widget_kwargs = {
'label': self.keywords_label,
'line_name': self.line_name,
'required': True
}
self.keywords = RepeatingElement(
add_text='Add keyword',
remove_text='Remove last',
widget_kwargs=widget_kwargs,
)
self.keywords.ui.italic_label.setStyleSheet('')
self.keywords.add_another()
self.ui.keywords_layout.insertWidget(0, self.keywords)
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_vertdef.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.altres_list = RepeatingElement(widget_kwargs={
'label': 'Altitude Resolution',
'line_name': 'fgdc_altres',
'required': True,
},
add_text='+',
remove_text='-')
self.altres_list.add_another()
self.ui.altsys_contents.layout().insertWidget(1, self.altres_list)
self.depthres_list = RepeatingElement(widget_kwargs={
'label': 'Depth Resolution',
'line_name': 'fgdc_depthres',
'required': True,
},
add_text='+',
remove_text='-')
self.depthres_list.add_another()
self.ui.depthsys_contents.layout().insertWidget(1, self.depthres_list)
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_crossref.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.crossrefs = RepeatingElement(
which="tab",
tab_label="Crossref",
add_text=" Add Additional Crossref ",
widget=Crossref,
remove_text=" Remove Selected Crossref ",
italic_text="",
)
self.crossrefs.add_another()
self.ui.crossref_widget.layout().addWidget(self.crossrefs)
self.ui.crossref_widget.hide()
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_keywords_repeater.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.ui.thesaurus_label = self.thesaurus_label
widget_kwargs = {
"label": self.keywords_label,
"line_name": self.line_name,
"required": True,
"spellings": self.spellings,
}
self.keywords = RepeatingElement(
add_text="Add keyword",
remove_text="Remove last",
widget_kwargs=widget_kwargs,
)
self.keywords.ui.italic_label.setStyleSheet("")
self.keywords.add_another()
self.ui.keywords_layout.insertWidget(0, self.keywords)
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_theme_list.Ui_theme_list()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.ui.theme_tabs.setStyleSheet(
"QTabBar::tab::disabled {width: 0; height: 0; margin: 0; padding: 0; border: none;} "
)
self.iso_kws = RepeatingElement(
which="vertical",
add_text="Add additonal",
widget=IsoKeyword,
remove_text="Remove Keyword",
italic_text="ISO Topic Category Keywords",
)
self.iso_kws.add_another()
self.ui.iso_keywords_layout.addWidget(self.iso_kws)
self.thesauri = []
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_timeperd.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.single_date = FGDCDate(label=' Single Date ',
fgdc_name='fgdc_caldate')
self.ui.fgdc_sngdate.layout().insertWidget(0, self.single_date)
self.range_start_date = FGDCDate(label='Start ',
fgdc_name='fgdc_begdate')
self.range_end_date = FGDCDate(label='End ', fgdc_name='fgdc_enddate')
self.ui.layout_daterange.addWidget(self.range_start_date)
self.ui.layout_daterange.addWidget(self.range_end_date)
date_widget_kwargs = {
'show_format': False,
'label': 'Individual Date ',
'fgdc_name': 'fgdc_caldate',
'parent_fgdc_name': 'fgdc_sngdate'
}
self.multi_dates = RepeatingElement(widget=FGDCDate,
widget_kwargs=date_widget_kwargs)
self.multi_dates.add_another()
self.switch_primary()
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_timeinfo.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.single_date = FGDCDate(label=" Single Date ",
fgdc_name="fgdc_caldate")
self.ui.fgdc_sngdate.layout().insertWidget(0, self.single_date)
self.range_start_date = FGDCDate(label="Start ",
fgdc_name="fgdc_begdate")
self.range_end_date = FGDCDate(label="End ", fgdc_name="fgdc_enddate")
self.ui.layout_daterange.addWidget(self.range_start_date)
self.ui.layout_daterange.addWidget(self.range_end_date)
date_widget_kwargs = {
"show_format": False,
"label": "Individual Date ",
"fgdc_name": "fgdc_caldate",
"parent_fgdc_name": "fgdc_sngdate",
}
self.multi_dates = RepeatingElement(widget=FGDCDate,
widget_kwargs=date_widget_kwargs)
self.multi_dates.add_another()
self.switch_primary()
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_vertdef.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.altres_list = RepeatingElement(
widget_kwargs={
"label": "Altitude Resolution",
"line_name": "fgdc_altres",
"required": True,
},
add_text="+",
remove_text="-",
)
self.altres_list.add_another()
self.ui.altsys_contents.layout().insertWidget(1, self.altres_list)
self.depthres_list = RepeatingElement(
widget_kwargs={
"label": "Depth Resolution",
"line_name": "fgdc_depthres",
"required": True,
},
add_text="+",
remove_text="-",
)
self.depthres_list.add_another()
self.ui.depthsys_contents.layout().insertWidget(1, self.depthres_list)
def build_ui(self, ):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_citeinfo.Ui_parent_form()
self.ui.setupUi(self)
if self.include_lwork:
self.lworkcit_widget = Citeinfo(parent=self, include_lwork=False)
self.lworkcit_widget.ui.lbl_dataset_title.setText(
'Larger Work Title')
self.ui.lworkcite_widget.layout().addWidget(self.lworkcit_widget)
self.lworkcit_widget.ui.fgdc_geoform.setEditText('publication')
else:
self.ui.fgdc_lworkcit.hide()
self.include_lworkext_change(self.ui.radio_lworkyes.isChecked())
self.ui.series_ext.hide()
self.ui.pub_ext.hide()
self.ui.pubdate_widget = FGDCDate(label='YYYYMMDD ',
show_format=False,
required=True,
fgdc_name='fgdc_pubdate')
self.ui.pubdate_layout.addWidget(self.ui.pubdate_widget)
self.onlink_list = RepeatingElement(add_text='Add online link',
remove_text='Remove last',
widget_kwargs={
'label': 'Link',
'line_name': 'fgdc_onlink'
})
self.onlink_list.add_another()
self.ui.onlink_layout.addWidget(self.onlink_list)
self.fgdc_origin = RepeatingElement(
add_text='Add originator',
remove_text='Remove last',
widget_kwargs={
'label': 'Originator',
'line_name': 'fgdc_origin',
'required': True,
'placeholder_text': self.origin_hint,
'spellings': False
},
)
self.ui.originator_layout.addWidget(self.fgdc_origin)
self.fgdc_origin.add_another()
self.setup_dragdrop(self)
if not hananero_installed:
self.ui.btn_import_doi.hide()
class KeywordsRepeater(WizardWidget):
drag_label = "NA <NA>"
def __init__(self, thesaurus_label='Thesaurus', keywords_label='Keywords:',
line_name='kw', parent=None):
self.thesaurus_label = thesaurus_label
self.keywords_label = keywords_label
self.line_name = line_name
self.keywords = None
WizardWidget.__init__(self, parent=parent)
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_keywords_repeater.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.ui.thesaurus_label = self.thesaurus_label
widget_kwargs = {'label': self.keywords_label,
'line_name': self.line_name,
'required': True}
self.keywords = RepeatingElement(add_text='Add keyword',
remove_text='Remove last',
widget_kwargs=widget_kwargs)
self.keywords.ui.italic_label.setStyleSheet('')
self.keywords.add_another()
self.ui.keywords_layout.insertWidget(0, self.keywords)
def lock(self):
self.ui.fgdc_themekt.setReadOnly(True)
self.keywords.ui.addAnother.setEnabled(False)
def get_keywords(self):
return [kw.added_line.text() for kw in self.keywords.get_widgets()]
def add_another(self, locked=False):
widget = self.keywords.add_another()
widget.setObjectName(self.line_name)
widget.added_line.setReadOnly(locked)
return widget
def get_widgets(self):
return self.keywords.get_widgets()
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_ProcessStep.Ui_Form()
self.ui.setupUi(self)
self.ui.fgdc_procdesc.heightMin = 150
self.setup_dragdrop(self)
self.single_date = FGDCDate(show_format=False,
required=True,
label="",
fgdc_name="fgdc_procdate")
self.proccont = ProcessContact()
self.ui.fgdc_procdate.setLayout(QVBoxLayout())
self.ui.fgdc_procdate.layout().insertWidget(0, self.single_date)
self.ui.widget_proccont.layout().insertWidget(0, self.proccont)
self.srcused_list = RepeatingElement(
add_text="Add another",
remove_text="Remove last",
widget_kwargs={
"label": "Source used",
"line_name": "fgdc_srcused"
},
)
self.srcused_list.add_another()
self.ui.srcused_groupbox.layout().addWidget(self.srcused_list)
self.srcprod_list = RepeatingElement(
add_text="Add another",
remove_text="Remove last",
widget_kwargs={
"label": "Source produced:",
"line_name": "fgdc_srcprod"
},
)
self.srcprod_list.add_another()
self.ui.srcprod_groupbox.layout().addWidget(self.srcprod_list)
self.clear_widget()
def build_ui(self, ):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_Citation.Ui_fgdc_citation()
self.ui.setupUi(self)
if self.include_lwork:
self.lworkcit_widget = Citation(parent=self, include_lwork=False)
self.ui.lworkcite_widget.layout().addWidget(self.lworkcit_widget)
else:
self.ui.fgdc_lworkcit.hide()
self.include_lworkext_change(self.ui.radio_lworkyes.isChecked())
self.ui.series_ext.hide()
self.ui.pub_ext.hide()
self.ui.pubdate_widget = SingleDate(label='YYYMMDD ',
show_format=False,
required=True)
self.ui.pubdate_layout.addWidget(self.ui.pubdate_widget)
self.onlink_list = RepeatingElement(add_text='Add online link',
remove_text='Remove last',
widget_kwargs={
'label': 'Link',
'line_name': 'fgdc_onlink'
})
self.onlink_list.add_another()
self.ui.onlink_layout.addWidget(self.onlink_list)
self.fgdc_origin = RepeatingElement(add_text='Add originator',
remove_text='Remove last',
widget_kwargs={
'label': 'Originator',
'line_name': 'fgdc_origin',
'required': True
})
self.fgdc_origin.add_another()
self.ui.originator_layout.addWidget(self.fgdc_origin)
self.setup_dragdrop(self)
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_procstep.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.proc_step = RepeatingElement(which='tab',
tab_label='Step', add_text='Additional Step',
widget=ProcessStep, remove_text='Remove Step', italic_text='Describe the methods performed to collect or generate the data.\n Provide as much detail as possible.')
self.proc_step.add_another()
self.ui.widget_procstep.layout().addWidget(self.proc_step)
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_procstep.Ui_Form()#.Ui_USGSContactInfoWidgetMain()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.proc_step = RepeatingElement(which='tab',
tab_label='Step', add_text='Additional Step',
widget=ProcessStep, remove_text='Remove Step', italic_text='Processing Steps Taken')
#self.proc_step = RepeatingElement(params=params, which='tab', tab_label='Source',)
self.proc_step.add_another()
self.ui.widget_procstep.layout().addWidget(self.proc_step)
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_taxoncl.Ui_fgdc_taxoncl()
self.ui.setupUi(self)
widget_kwargs = {'line_name': 'common', 'required': False}
self.commons = RepeatingElement(add_text='Add Common',
remove_text='Remove last',
widget_kwargs=widget_kwargs,
show_buttons=False)
self.commons.add_another()
self.ui.horizontalLayout_4.addWidget(self.commons)
self.child_taxoncl = []
self.setup_dragdrop(self)
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_sourceinput.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.src_info = RepeatingElement(which='tab',
tab_label='Source',
add_text='Source Input',
widget=SRCInfo,
remove_text='Remove Source',
italic_text='Source')
self.src_info.add_another()
self.ui.frame_sourceinfo.layout().addWidget(self.src_info)
self.ui.frame_sourceinfo.hide()
class SourceInput(WizardWidget):
drag_label = "Source Information <srcinfo>"
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_sourceinput.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.src_info = RepeatingElement(which='tab',
tab_label='Source',
add_text='Source Input',
widget=SRCInfo,
remove_text='Remove Source',
italic_text='Source')
self.src_info.add_another()
self.ui.frame_sourceinfo.layout().addWidget(self.src_info)
self.ui.frame_sourceinfo.hide()
def connect_events(self):
"""
Connect the appropriate GUI components with the corresponding functions
Returns
-------
None
"""
self.ui.radio_sourceyes.toggled.connect(self.include_sources_change)
def include_sources_change(self, b):
"""
Extended citation to support a larger body of information for the record.
Parameters
----------
b: qt event
Returns
-------
None
"""
if b:
self.ui.frame_sourceinfo.show()
else:
self.ui.frame_sourceinfo.hide()
def dragEnterEvent(self, e):
"""
Only accept Dragged items that can be converted to an xml object with
a root tag called 'accconst'
Parameters
----------
e : qt event
Returns
-------
None
"""
print("pc drag enter")
mime_data = e.mimeData()
if e.mimeData().hasFormat('text/plain'):
parser = etree.XMLParser(ns_clean=True,
recover=True,
encoding='utf-8')
element = etree.fromstring(mime_data.text(), parser=parser)
if element.tag == 'lineage':
e.accept()
else:
e.ignore()
def clear_widget(self):
self.ui.radio_sourceno_2.setChecked(True)
WizardWidget.clear_widget(self)
def _to_xml(self):
"""
encapsulates the text in an element tree representing Sources Input
Returns
-------
series of srcinfo element tag in lineage xml tree
"""
lineage = etree.Element('lineage')
if self.ui.radio_sourceyes.isChecked():
print("in to xml")
cnt = 0
srcinfo_list = self.src_info.get_widgets()
for srcinfo in srcinfo_list:
lineage.append(srcinfo._to_xml())
# while cnt < len(list_widgets):
# #lineage.append(list_widgets[cnt]._to_xml())
# lineage.append(SRCInfo._to_xml(list_widgets[cnt]))
# cnt += 1
return lineage
# elif self.ui.radio_sourceno.isChecked():
# pass
def _from_xml(self, xml_srcinput):
"""
parses the xml code into the relevant accconst elements
Parameters
----------
access_constraints - the xml element status and its contents
Returns
-------
None
"""
try:
if xml_srcinput.tag == 'lineage':
self.src_info.clear_widgets(add_another=False)
self.ui.frame_sourceinfo.show()
self.ui.radio_sourceyes.setChecked(True)
xml_srcinput = xml_srcinput.findall('srcinfo')
if xml_srcinput:
for srcinput in xml_srcinput:
srcinfo_widget = self.src_info.add_another()
srcinfo_widget._from_xml(srcinput)
else:
self.ui.radio_sourceno_2.setChecked(True)
self.src_info.add_another()
except KeyError:
pass
class ThemeList(WizardWidget): #
drag_label = "Theme Keywords <keywords>"
acceptable_tags = ["keywords"]
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_theme_list.Ui_theme_list()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.ui.theme_tabs.setStyleSheet(
"QTabBar::tab::disabled {width: 0; height: 0; margin: 0; padding: 0; border: none;} "
)
self.iso_kws = RepeatingElement(
which="vertical",
add_text="Add additonal",
widget=IsoKeyword,
remove_text="Remove Keyword",
italic_text="ISO Topic Category Keywords",
)
self.iso_kws.add_another()
self.ui.iso_keywords_layout.addWidget(self.iso_kws)
self.thesauri = []
def connect_events(self):
"""
Connect the appropriate GUI components with the corresponding functions
Returns
-------
None
"""
self.ui.btn_add_thesaurus.clicked.connect(self.add_another)
self.ui.btn_remove_selected.clicked.connect(self.remove_selected)
self.ui.btn_add_iso.clicked.connect(self.add_iso)
self.ui.btn_search_controlled.clicked.connect(self.search_controlled)
def add_another(self, click=False, tab_label="", locked=False):
if (
"None" not in [t.get_thesaurus_name() for t in self.thesauri]
and tab_label == ""
):
theme_widget = self.add_keyword(keyword="", thesaurus="None", locked=False)
else:
theme_widget = Theme()
theme_widget.ui.fgdc_themekt.textChanged.connect(self.changed_thesaurus)
self.ui.theme_tabs.addTab(theme_widget, tab_label)
self.ui.theme_tabs.setCurrentIndex(self.ui.theme_tabs.count() - 1)
self.thesauri.append(theme_widget)
return theme_widget
def changed_thesaurus(self, s):
current_index = self.ui.theme_tabs.currentIndex()
self.ui.theme_tabs.setTabText(current_index, "Thesaurus: " + s)
def remove_selected(self):
current_index = self.ui.theme_tabs.currentIndex()
if current_index == 0:
self.remove_iso()
else:
self.ui.theme_tabs.removeTab(current_index)
del self.thesauri[current_index - 1]
def remove_iso(self):
self.ui.theme_tabs.setTabEnabled(0, False)
self.ui.fgdc_theme.hide()
def add_iso(self):
self.ui.theme_tabs.setTabEnabled(0, True)
self.ui.theme_tabs.setTabText(0, "ISO 19115")
self.ui.fgdc_theme.show()
self.ui.theme_tabs.setCurrentIndex(0)
self.resize(100, 100)
def get_children(self, widget):
children = []
if self.ui.theme_tabs.isTabEnabled(0):
children.append(self.ui.fgdc_theme)
for theme in self.thesauri:
children.append(theme)
return children
def clear_widget(self, remove_iso=False):
self.iso_kws.clear_widgets()
if remove_iso:
self.remove_iso()
for i in range(len(self.thesauri), -1, -1):
self.ui.theme_tabs.setCurrentIndex(i)
self.remove_selected()
self.ui.fgdc_themekt.setText("ISO 19115 Topic Category")
def search_controlled(self):
self.thesaurus_search = ThesaurusSearch.ThesaurusSearch(
add_term_function=self.add_keyword, parent=self
)
self.thesaurus_search.setWindowTitle("Search USGS Controlled Vocabularies")
self.thesaurus_search.show()
def add_keyword(self, keyword=None, thesaurus=None, locked=True):
theme_widget = None
for i, theme in enumerate(self.thesauri):
if theme.ui.fgdc_themekt.text() == thesaurus:
theme_widget = theme
try:
self.ui.theme_tabs.setCurrentIndex(i + 1)
except IndexError:
pass
if theme_widget is None:
shortname = thesaurus.split(" ")[0]
theme_widget = self.add_another(tab_label=shortname, locked=locked)
theme_widget.ui.fgdc_themekt.setText(thesaurus)
if locked:
theme_widget.lock()
self.changed_thesaurus(shortname)
theme_widget.add_keyword(keyword, locked=locked)
return theme_widget
def to_xml(self):
"""
encapsulates the QPlainTextEdit text in an element tag
Returns
-------
procstep element tag in xml tree
"""
keywords = xml_utils.xml_node("keywords")
if self.ui.theme_tabs.isTabEnabled(0):
theme = xml_utils.xml_node("theme", parent_node=keywords)
themekt = xml_utils.xml_node(
"themekt", text=self.ui.fgdc_themekt.text(), parent_node=theme
)
for isokw in self.iso_kws.get_widgets():
themekey = xml_utils.xml_node(
"themekey",
text=isokw.ui.fgdc_themekey.currentText(),
parent_node=theme,
)
for theme in self.thesauri:
theme_xml = theme.to_xml()
if (
theme_xml.xpath("themekt")[0].text == "None"
and len(theme_xml.xpath("themekey")) == 1
and theme_xml.xpath("themekey")[0].text is None
and len(self.thesauri) >= 2
):
# Skip themes that still have the default thesaurus ('None')
# with no keywords entered,
# when they added an additional thesaurus
pass
else:
keywords.append(theme_xml)
return keywords
def from_xml(self, keywords_xml):
"""
parses the xml code into the relevant procstep elements
Parameters
----------
process_step - the xml element status and its contents
Returns
-------
None
"""
self.clear_widget(remove_iso=True)
self.original_xml = keywords_xml
if keywords_xml.tag == "keywords":
for theme_xml in xml_utils.search_xpath(keywords_xml, "theme", False):
themekt = xml_utils.get_text_content(theme_xml, "themekt")
if themekt is not None and "iso 19115" in themekt.lower():
self.add_iso()
self.iso_kws.clear_widgets(add_another=False)
for themekey in xml_utils.search_xpath(
theme_xml, "themekey", only_first=False
):
iso = self.iso_kws.add_another()
iso.ui.fgdc_themekey.setCurrentText(themekey.text)
else:
theme = self.add_another()
theme.from_xml(theme_xml)
class ProcStep(WizardWidget): #
drag_label = "Process Step <procstep>"
acceptable_tags = ['lineage']
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_procstep.Ui_Form() #.Ui_USGSContactInfoWidgetMain()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.proc_step = RepeatingElement(which='tab',
tab_label='Step',
add_text='Additional Step',
widget=ProcessStep,
remove_text='Remove Step',
italic_text='Processing Steps Taken')
#self.proc_step = RepeatingElement(params=params, which='tab', tab_label='Source',)
self.proc_step.add_another()
self.ui.widget_procstep.layout().addWidget(self.proc_step)
def _to_xml(self):
"""
encapsulates the etree process step in an element tag
Returns
-------
procstep portion of the lineageg element tag in xml tree
"""
lineage = etree.Element('lineage')
procstep_list = self.proc_step.get_widgets()
for procstep in procstep_list:
lineage.append(procstep._to_xml())
return lineage
def _from_xml(self, xml_procstep):
"""
parses the xml code into the relevant accconst elements
Parameters
----------
access_constraints - the xml element status and its contents
Returns
-------
None
"""
try:
if xml_procstep.tag == 'lineage':
self.proc_step.clear_widgets(add_another=False)
xml_procstep = xml_procstep.findall('procstep')
if xml_procstep:
for procstep in xml_procstep:
procdesc_widget = self.proc_step.add_another()
procdesc_widget._from_xml(procstep)
else:
self.proc_step.add_another()
except KeyError:
pass
class Crossref_list(WizardWidget):
drag_label = "Cross Reference <crossref>"
acceptable_tags = ['idinfo']
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_crossref.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.crossrefs = RepeatingElement(which='tab',
tab_label='Crossref', add_text=' Add Additional Crossref ',
widget=Crossref, remove_text=' Remove Selected Crossref ', italic_text='')
self.crossrefs.add_another()
self.ui.crossref_widget.layout().addWidget(self.crossrefs)
self.ui.crossref_widget.hide()
def connect_events(self):
"""
Connect the appropriate GUI components with the corresponding functions
Returns
-------
None
"""
self.ui.radio_crossrefyes.toggled.connect(self.crossref_used_change)
def crossref_used_change(self, b):
if b:
self.ui.crossref_widget.show()
else:
self.ui.crossref_widget.hide()
def has_content(self):
return self.ui.radio_crossrefyes.isChecked()
def get_children(self, widget=None):
children = []
for crossref in self.crossrefs.get_widgets():
children.append(crossref)
return children
def clear_widget(self):
super(Crossref_list, self).clear_widget()
self.ui.radio_crossrefno.setChecked(True)
def to_xml(self):
"""
encapsulates the etree process step in an element tag
Returns
-------
procstep portion of the lineageg element tag in xml tree
"""
crossrefs = []
idinfo = xml_utils.xml_node(tag='idinfo')
for crossref in self.crossrefs.get_widgets():
idinfo.append(crossref.to_xml())
return idinfo
def from_xml(self, xml_idinfo):
"""
parses the xml code into the relevant accconst elements
Parameters
----------
access_constraints - the xml element status and its contents
Returns
-------
None
"""
try:
if xml_idinfo.tag == 'idinfo':
self.crossrefs.clear_widgets(add_another=False)
crossrefs = xml_utils.search_xpath(xml_idinfo, 'crossref', only_first=False)
if crossrefs:
self.ui.radio_crossrefyes.setChecked(True)
else:
self.crossrefs.add_another()
self.ui.radio_crossrefno.setChecked(True)
for crossref in crossrefs:
crossref_widget = self.crossrefs.add_another()
crossref_widget.from_xml(xml_utils.search_xpath(crossref, 'citeinfo'))
except KeyError:
self.ui.radio_crossrefno.setChecked(True)
class ProcessStep(WizardWidget): #
drag_label = "Process Step <procstep>"
acceptable_tags = ['procstep']
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_ProcessStep.Ui_Form()
self.ui.setupUi(self)
self.ui.fgdc_procdesc.heightMin = 150
self.setup_dragdrop(self)
self.single_date = FGDCDate(show_format=False,
required=True,
label='',
fgdc_name='fgdc_procdate')
self.proccont = ProcessContact()
self.ui.fgdc_procdate.setLayout(QVBoxLayout())
self.ui.fgdc_procdate.layout().insertWidget(0, self.single_date)
self.ui.widget_proccont.layout().insertWidget(0, self.proccont)
self.srcused_list = RepeatingElement(add_text='Add another',
remove_text='Remove last',
widget_kwargs={
'label': 'Source used',
'line_name': 'fgdc_srcused'
})
self.srcused_list.add_another()
self.ui.srcused_groupbox.layout().addWidget(self.srcused_list)
self.srcprod_list = RepeatingElement(add_text='Add another',
remove_text='Remove last',
widget_kwargs={
'label': 'Source produced:',
'line_name': 'fgdc_srcprod'
})
self.srcprod_list.add_another()
self.ui.srcprod_groupbox.layout().addWidget(self.srcprod_list)
self.clear_widget()
def clear_widget(self):
super(self.__class__, self).clear_widget()
self.proccont.ui.rbtn_no.setChecked(True)
def to_xml(self):
"""
encapsulates the QPlainTextEdit text in an element tag
Returns
-------
procstep element tag in xml tree
"""
procstep = xml_utils.xml_node(tag='procstep')
procdesc = xml_utils.xml_node(tag='procdesc')
procdesc.text = self.findChild(QPlainTextEdit,
"fgdc_procdesc").toPlainText()
procstep.append(procdesc)
for srcused in self.srcused_list.get_widgets():
if srcused.text():
xml_utils.xml_node('srcused',
text=srcused.text(),
parent_node=procstep)
procdate = xml_utils.xml_node(tag='procdate')
date_var = self.single_date.findChild(QLineEdit,
"fgdc_procdate").text()
procdate.text = date_var
procstep.append(procdate)
for srcprod in self.srcprod_list.get_widgets():
if srcprod.text():
xml_utils.xml_node('srcprod',
text=srcprod.text(),
parent_node=procstep)
if self.proccont.ui.rbtn_yes.isChecked():
proccont = self.proccont.to_xml()
procstep.append(proccont)
return procstep
def from_xml(self, xml_processstep):
"""
parses the xml code into the relevant procstep elements
Parameters
----------
process_step - the xml element status and its contents
Returns
-------
None
"""
try:
if xml_processstep.tag == 'procstep':
utils.populate_widget(self, xml_processstep)
if xml_processstep.xpath('procdate'):
self.single_date.set_date(
xml_processstep.xpath('procdate')[0].text)
if xml_processstep.xpath('proccont'):
self.proccont.ui.rbtn_yes.setChecked(True)
cntinfo_node = xml_processstep.xpath('proccont/cntinfo')[0]
self.proccont.from_xml(cntinfo_node)
else:
self.proccont.ui.rbtn_no.setChecked(True)
pass
srcuseds = xml_processstep.findall("srcused")
if srcuseds:
self.srcused_list.clear_widgets(add_another=False)
for srcused in srcuseds:
srcused_widget = self.srcused_list.add_another()
srcused_widget.setText(srcused.text)
else:
self.srcused_list.clear_widgets(add_another=True)
srcprods = xml_processstep.findall("srcprod")
if srcprods:
self.srcprod_list.clear_widgets(add_another=False)
for srcprod in srcprods:
srcprod_widget = self.srcprod_list.add_another()
srcprod_widget.setText(srcprod.text)
else:
self.srcprod_list.clear_widgets(add_another=True)
else:
print("The tag is not procstep")
except KeyError:
pass
class Taxoncl(WizardWidget):
drag_label = "taxon class <taxoncl>"
acceptable_tags = ["taxoncl"]
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_taxoncl.Ui_fgdc_taxoncl()
self.ui.setupUi(self)
widget_kwargs = {"line_name": "common", "required": False}
self.commons = RepeatingElement(
add_text="Add Common",
remove_text="Remove last",
widget_kwargs=widget_kwargs,
show_buttons=False,
)
self.commons.add_another()
self.ui.horizontalLayout_4.addWidget(self.commons)
self.child_taxoncl = []
self.setup_dragdrop(self)
def clear_widget(self):
"""
Clears all content from this widget
Returns
-------
None
"""
self.ui.fgdc_taxonrn.clear()
self.ui.fgdc_taxonrv.clear()
self.commons.clear_widgets()
for taxoncl in self.child_taxoncl:
taxoncl.deleteLater()
self.child_taxoncl = []
def to_xml(self):
"""
encapsulates the QTabWidget text for Metadata Time in an element tag
Returns
-------
timeperd element tag in xml tree
"""
taxoncl = xml_utils.xml_node("taxoncl")
taxonrn = xml_utils.xml_node("taxonrn",
text=self.ui.fgdc_taxonrn.text(),
parent_node=taxoncl)
taxonrv = xml_utils.xml_node("taxonrv",
text=self.ui.fgdc_taxonrv.text(),
parent_node=taxoncl)
common_names = [c.text() for c in self.commons.get_widgets()]
for common_name in common_names:
if common_name:
common = xml_utils.xml_node("common",
text=common_name,
parent_node=taxoncl)
for child_taxoncl in self.child_taxoncl:
taxoncl.append(child_taxoncl.to_xml())
return taxoncl
def from_xml(self, taxoncl):
"""
parses the xml code into the relevant timeperd elements
Parameters
----------
metadata_date - the xml element timeperd and its contents
Returns
-------
None
"""
try:
if taxoncl.tag == "taxoncl":
self.ui.fgdc_taxonrn.setText(taxoncl.xpath("taxonrn")[0].text)
self.ui.fgdc_taxonrv.setText(taxoncl.xpath("taxonrv")[0].text)
commons = xml_utils.search_xpath(taxoncl,
"common",
only_first=False)
if commons:
self.commons.clear_widgets(add_another=False)
for common in commons:
this_common = self.commons.add_another()
this_common.setText(common.text)
children_taxoncl = taxoncl.xpath("taxoncl")
for child_taxoncl in children_taxoncl:
child_widget = Taxoncl()
child_widget.from_xml(child_taxoncl)
self.ui.child_taxoncl.layout().addWidget(child_widget)
self.child_taxoncl.append(child_widget)
else:
print("The tag is not a detailed")
except KeyError:
pass
class ProcStep(WizardWidget): #
drag_label = "Process Step <procstep>"
acceptable_tags = ['lineage']
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_procstep.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.proc_step = RepeatingElement(which='tab',
tab_label='Step', add_text='Additional Step',
widget=ProcessStep, remove_text='Remove Step', italic_text='Describe the methods performed to collect or generate the data.\n Provide as much detail as possible.')
self.proc_step.add_another()
self.ui.widget_procstep.layout().addWidget(self.proc_step)
def to_xml(self):
"""
encapsulates the etree process step in an element tag
Returns
-------
procstep portion of the lineageg element tag in xml tree
"""
lineage = xml_utils.xml_node(tag='lineage')
procstep_list = self.proc_step.get_widgets()
for procstep in procstep_list:
lineage.append(procstep.to_xml())
return lineage
def from_xml(self, xml_procstep):
"""
parses the xml code into the relevant accconst elements
Parameters
----------
access_constraints - the xml element status and its contents
Returns
-------
None
"""
try:
if xml_procstep.tag == 'lineage':
self.proc_step.clear_widgets(add_another=False)
xml_procstep = xml_procstep.findall('procstep')
if xml_procstep:
for procstep in xml_procstep:
procdesc_widget = self.proc_step.add_another()
procdesc_widget.from_xml(procstep)
else:
self.proc_step.add_another()
except KeyError:
pass
def build_ui(self, ):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_citeinfo.Ui_parent_form()
self.ui.setupUi(self)
if self.include_lwork:
self.lworkcit_widget = Citeinfo(parent=self, include_lwork=False)
self.lworkcit_widget.ui.lbl_dataset_title.setText(
"Larger Work Title")
self.ui.lworkcite_widget.layout().addWidget(self.lworkcit_widget)
self.lworkcit_widget.ui.fgdc_geoform.setEditText("publication")
else:
self.ui.fgdc_lworkcit.hide()
self.include_lworkext_change(self.ui.radio_lworkyes.isChecked())
self.ui.series_ext.hide()
self.ui.pub_ext.hide()
self.ui.pubdate_widget = FGDCDate(
label="YYYYMMDD ",
show_format=False,
required=True,
fgdc_name="fgdc_pubdate",
)
self.ui.pubdate_layout.addWidget(self.ui.pubdate_widget)
self.onlink_list = RepeatingElement(
add_text="Add online link",
remove_text="Remove last",
italic_text=
"Is there a link to the data or the agency that produced it? if so, provide the URL(s) ",
widget_kwargs={
"label": "Link",
"line_name": "fgdc_onlink"
},
)
self.onlink_list.add_another()
self.ui.onlink_layout.addWidget(self.onlink_list)
self.fgdc_origin = RepeatingElement(
add_text="Add originator",
remove_text="Remove last",
italic_text=
"Who created the dataset? List the organization and/or person(s)",
widget_kwargs={
"label": "Originator",
"line_name": "fgdc_origin",
"required": True,
"placeholder_text": self.origin_hint,
"spellings": False,
},
)
self.ui.originator_layout.addWidget(self.fgdc_origin)
self.fgdc_origin.add_another()
self.setup_dragdrop(self)
if not hananero_installed:
self.ui.btn_import_doi.hide()
class Citeinfo(WizardWidget): #
drag_label = "Citation information <citeinfo>"
acceptable_tags = ['citation', 'citeinfo']
def __init__(self, parent=None, include_lwork=True):
self.include_lwork = include_lwork
self.schema = 'bdp'
WizardWidget.__init__(self, parent=parent)
def build_ui(self, ):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_citeinfo.Ui_parent_form()
self.ui.setupUi(self)
if self.include_lwork:
self.lworkcit_widget = Citeinfo(parent=self, include_lwork=False)
self.lworkcit_widget.ui.lbl_dataset_title.setText('Larger Work Title')
self.ui.lworkcite_widget.layout().addWidget(self.lworkcit_widget)
self.lworkcit_widget.ui.fgdc_geoform.setEditText('publication')
else:
self.ui.fgdc_lworkcit.hide()
self.include_lworkext_change(self.ui.radio_lworkyes.isChecked())
self.ui.series_ext.hide()
self.ui.pub_ext.hide()
self.ui.pubdate_widget = FGDCDate(label='YYYYMMDD ',
show_format=False, required=True,
fgdc_name='fgdc_pubdate')
self.ui.pubdate_layout.addWidget(self.ui.pubdate_widget)
self.onlink_list = RepeatingElement(add_text='Add online link',
remove_text='Remove last',
widget_kwargs={'label': 'Link',
'line_name':'fgdc_onlink'})
self.onlink_list.add_another()
self.ui.onlink_layout.addWidget(self.onlink_list)
self.fgdc_origin = RepeatingElement(add_text='Add originator',
remove_text='Remove last',
widget_kwargs={'label': 'Originator',
'line_name':'fgdc_origin',
'required':True})
self.fgdc_origin.add_another()
self.ui.originator_layout.addWidget(self.fgdc_origin)
self.setup_dragdrop(self)
def connect_events(self):
"""
Connect the appropriate GUI components with the corresponding functions
Returns
-------
None
"""
self.ui.radio_lworkyes.toggled.connect(self.include_lworkext_change)
self.ui.radio_seriesyes.toggled.connect(self.include_seriesext_change)
self.ui.radio_pubinfoyes.toggled.connect(self.include_pubext_change)
def include_seriesext_change(self, b):
"""
Extended citation to support series information of the record.
Parameters
----------
b: qt event
Returns
-------
None
"""
if b:
self.ui.series_ext.show()
else:
self.ui.series_ext.hide()
def include_pubext_change(self, b):
"""
Extended citation to support publication information of the record.
Parameters
----------
b: qt event
Returns
-------
None
"""
if b:
self.ui.pub_ext.show()
else:
self.ui.pub_ext.hide()
def include_lworkext_change(self, b):
"""
Extended citation to support a larger body of information for the record.
Parameters
----------
b: qt event
Returns
-------
None
"""
if b:
self.ui.lworkcite_widget.show()
else:
self.ui.lworkcite_widget.hide()
def _to_xml(self):
"""
encapsulates the QLineEdit text in an element tag
Returns
-------
citation element tag in xml tree
"""
citeinfo = xml_utils.xml_node('citeinfo')
for origin in self.fgdc_origin.get_widgets():
xml_utils.xml_node('origin', text=origin.added_line.text(),
parent_node=citeinfo)
pubdate = xml_utils.xml_node("pubdate",
text=self.ui.pubdate_widget.get_date(),
parent_node=citeinfo)
title = xml_utils.xml_node("title", self.ui.fgdc_title.text(),
parent_node=citeinfo)
geoform = xml_utils.xml_node("geoform",
self.ui.fgdc_geoform.currentText(),
parent_node=citeinfo)
if self.ui.radio_seriesyes.isChecked():
serinfo = xml_utils.xml_node('serinfo', parent_node=citeinfo)
sername = xml_utils.xml_node('sername',
text=self.ui.fgdc_sername.text(),
parent_node=serinfo)
issue = xml_utils.xml_node('issue', text=self.ui.fgdc_issue.text(),
parent_node=serinfo)
if self.ui.radio_pubinfoyes.isChecked():
pubinfo = xml_utils.xml_node('pubinfo', parent_node=citeinfo)
pubplace = xml_utils.xml_node('pubplace', parent_node=pubinfo,
text=self.ui.fgdc_pubplace.text())
publish = xml_utils.xml_node('publish', parent_node=pubinfo,
text=self.ui.fgdc_publish.text())
for onlink in self.onlink_list.get_widgets():
if onlink.added_line.text() != '':
onlink_node = xml_utils.xml_node('onlink',
parent_node=citeinfo,
text=onlink.added_line.text())
if self.include_lwork and self.ui.radio_lworkyes.isChecked():
lworkcit = xml_utils.xml_node('lworkcit', parent_node=citeinfo)
lwork = self.lworkcit_widget._to_xml()
lworkcit.append(lwork)
return citeinfo
def _from_xml(self, citeinfo):
"""
parses the xml code into the relevant citation elements
Parameters
----------
citation - the xml element status and its contents
Returns
-------
None
"""
try:
if citeinfo.tag == "citation":
citeinfo = citeinfo.xpath('citeinfo')[0]
elif citeinfo.tag != 'citeinfo':
print("The tag is not 'citation' or 'citeinfo'")
return
self.fgdc_origin.clear_widgets()
originators = citeinfo.findall("origin")
if originators :
self.fgdc_origin.clear_widgets(add_another=False)
for origin in citeinfo.findall('origin'):
origin_widget = self.fgdc_origin.add_another()
origin_widget.added_line.setText(origin.text)
else:
self.fgdc_origin.add_another()
if citeinfo.findall('geoform'):
self.ui.fgdc_geoform.setEditText(citeinfo.findall('geoform')[0].text)
utils.populate_widget_element(self.ui.pubdate_widget.ui.fgdc_caldate,
citeinfo, 'pubdate')
utils.populate_widget_element(self.ui.fgdc_title, citeinfo, 'title')
self.onlink_list.clear_widgets()
if citeinfo.findall("onlink"):
for onlink in citeinfo.findall('onlink'):
onlink_widget = self.onlink_list.widgets[0]
onlink_widget.added_line.setText(onlink.text)
if citeinfo.xpath('serinfo'):
self.ui.radio_seriesyes.setChecked(True)
utils.populate_widget(self, citeinfo.xpath('serinfo')[0])
else:
self.ui.radio_seriesyes.setChecked(False)
pubinfo = citeinfo.xpath('pubinfo')
if pubinfo:
self.ui.radio_pubinfoyes.setChecked(True)
utils.populate_widget_element(self.ui.fgdc_publish, pubinfo[0], 'publish')
utils.populate_widget_element(self.ui.fgdc_pubplace, pubinfo[0], 'pubplace')
else:
self.ui.radio_pubinfoyes.setChecked(False)
self.ui.radioButton_8.setChecked(True)
if citeinfo.xpath('lworkcit'):
try:
self.ui.radio_lworkyes.setChecked(True)
self.lworkcit_widget._from_xml(citeinfo.xpath('lworkcit/citeinfo')[0])
except AttributeError:
msg = 'you pasted a citation element into the larger work citation area'
msg += '\n that contained a larger work citation'
msg += '\n Multiple nested larger work citations are not currently supported in the tool'
msg += '\n\n the larger work citation being pasted will be ignored'
QMessageBox.warning(self, "Dropped Content Warning", msg)
else:
self.ui.radio_lworkno.setChecked(True)
except KeyError:
pass
class Citation(WizardWidget): #
drag_label = "Citation <citation>"
def __init__(self, parent=None, include_lwork=True):
self.include_lwork = include_lwork
self.schema = 'bdp'
WizardWidget.__init__(self, parent=parent)
def build_ui(self, ):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_Citation.Ui_fgdc_citation()
self.ui.setupUi(self)
if self.include_lwork:
self.lworkcit_widget = Citation(parent=self, include_lwork=False)
self.ui.lworkcite_widget.layout().addWidget(self.lworkcit_widget)
else:
self.ui.fgdc_lworkcit.hide()
self.include_lworkext_change(self.ui.radio_lworkyes.isChecked())
self.ui.series_ext.hide()
self.ui.pub_ext.hide()
self.ui.pubdate_widget = SingleDate(label='YYYMMDD ',
show_format=False,
required=True)
self.ui.pubdate_layout.addWidget(self.ui.pubdate_widget)
self.onlink_list = RepeatingElement(add_text='Add online link',
remove_text='Remove last',
widget_kwargs={
'label': 'Link',
'line_name': 'fgdc_onlink'
})
self.onlink_list.add_another()
self.ui.onlink_layout.addWidget(self.onlink_list)
self.fgdc_origin = RepeatingElement(add_text='Add originator',
remove_text='Remove last',
widget_kwargs={
'label': 'Originator',
'line_name': 'fgdc_origin',
'required': True
})
self.fgdc_origin.add_another()
self.ui.originator_layout.addWidget(self.fgdc_origin)
self.setup_dragdrop(self)
def connect_events(self):
"""
Connect the appropriate GUI components with the corresponding functions
Returns
-------
None
"""
self.ui.radio_lworkyes.toggled.connect(self.include_lworkext_change)
self.ui.radio_seriesyes.toggled.connect(self.include_seriesext_change)
self.ui.radio_pubinfoyes.toggled.connect(self.include_pubext_change)
def include_seriesext_change(self, b):
"""
Extended citation to support series information of the record.
Parameters
----------
b: qt event
Returns
-------
None
"""
if b:
self.ui.series_ext.show()
else:
self.ui.series_ext.hide()
def include_pubext_change(self, b):
"""
Extended citation to support publication information of the record.
Parameters
----------
b: qt event
Returns
-------
None
"""
if b:
self.ui.pub_ext.show()
else:
self.ui.pub_ext.hide()
def include_lworkext_change(self, b):
"""
Extended citation to support a larger body of information for the record.
Parameters
----------
b: qt event
Returns
-------
None
"""
if b:
self.ui.lworkcite_widget.show()
else:
self.ui.lworkcite_widget.hide()
def dragEnterEvent(self, e):
"""
Only accept Dragged items that can be converted to an xml object with
a root tag called 'citation'
Parameters
----------
e : qt event
Returns
-------
None
"""
mime_data = e.mimeData()
if e.mimeData().hasFormat('text/plain'):
parser = etree.XMLParser(ns_clean=True,
recover=True,
encoding='utf-8')
element = etree.fromstring(mime_data.text(), parser=parser)
if element.tag in ['citation', 'citeinfo']:
e.accept()
else:
e.ignore()
def switch_schema(self, schema):
self.schema = schema
if schema == 'bdp':
self.ui.geoform_groupbox.show()
else:
self.ui.geoform_groupbox.hide()
if self.include_lwork:
self.lworkcit_widget.switch_schema(self.schema)
def _to_xml(self):
"""
encapsulates the QLineEdit text in an element tag
Returns
-------
citation element tag in xml tree
"""
citeinfo = xml_utils.xml_node('citeinfo')
for origin in self.fgdc_origin.get_widgets():
xml_utils.xml_node('origin',
text=origin.added_line.text(),
parent_node=citeinfo)
pubdate = xml_utils.xml_node("pubdate",
text=self.ui.pubdate_widget.get_date(),
parent_node=citeinfo)
title = xml_utils.xml_node("title",
self.ui.fgdc_title.text(),
parent_node=citeinfo)
if self.schema == 'bdp':
geoform = xml_utils.xml_node("geoform",
self.ui.fgdc_geoform.currentText(),
parent_node=citeinfo)
if self.ui.radio_seriesyes.isChecked():
serinfo = xml_utils.xml_node('serinfo', parent_node=citeinfo)
sername = xml_utils.xml_node('serinfo',
text=self.ui.fgdc_sername.text(),
parent_node=serinfo)
issue = xml_utils.xml_node('issue',
text=self.ui.fgdc_issue.text(),
parent_node=serinfo)
if self.ui.radio_pubinfoyes.isChecked():
pubinfo = xml_utils.xml_node('pubinfo', parent_node=citeinfo)
pubplace = xml_utils.xml_node('pubplace',
parent_node=pubinfo,
text=self.ui.fgdc_pubplace.text())
publish = xml_utils.xml_node('publish',
parent_node=pubinfo,
text=self.ui.fgdc_publish.text())
for onlink in self.onlink_list.get_widgets():
if onlink.added_line.text() != '':
onlink_node = xml_utils.xml_node('onlink',
parent_node=citeinfo,
text=onlink.added_line.text())
if self.include_lwork and self.ui.radio_lworkyes.isChecked():
lworkcit = xml_utils.xml_node('lworkcit', parent_node=citeinfo)
lwork = self.lworkcit_widget._to_xml()
lworkcit.append(lwork)
return citeinfo
def _from_xml(self, citeinfo):
"""
parses the xml code into the relevant citation elements
Parameters
----------
citation - the xml element status and its contents
Returns
-------
None
"""
try:
if citeinfo.tag == "citation":
citeinfo = citeinfo.xpath('citeinfo')[0]
elif citeinfo.tag != 'citeinfo':
print("The tag is not 'citation' or 'citeinfo'")
return
self.fgdc_origin.clear_widgets()
if citeinfo.findall("origin"):
for origin in citeinfo.findall('origin'):
origin_widget = self.fgdc_origin.widgets[0]
origin_widget.added_line.setText(origin.text)
else:
self.fgdc_origin.add_another()
if citeinfo.findall('geoform'):
self.ui.fgdc_geoform.setEditText(
citeinfo.findall('geoform')[0].text)
utils.populate_widget_element(
self.ui.pubdate_widget.ui.fgdc_caldate, citeinfo, 'pubdate')
utils.populate_widget_element(self.ui.fgdc_title, citeinfo,
'title')
self.onlink_list.clear_widgets()
if citeinfo.findall("onlink"):
for onlink in citeinfo.findall('onlink'):
onlink_widget = self.onlink_list.widgets[0]
onlink_widget.added_line.setText(onlink.text)
if citeinfo.xpath('serinfo'):
self.ui.radio_seriesyes.setChecked(True)
utils.populate_widget(self.ui.fgdc_serinfo,
citeinfo.xpath('serinfo')[0])
else:
self.ui.radio_seriesyes.setChecked(False)
pubinfo = citeinfo.xpath('pubinfo')
if pubinfo:
self.ui.radio_pubinfoyes.setChecked(True)
utils.populate_widget_element(self.ui.fgdc_publish, pubinfo[0],
'publish')
utils.populate_widget_element(self.ui.fgdc_pubplace,
pubinfo[0], 'pubplace')
else:
self.ui.radio_pubinfoyes.setChecked(False)
if citeinfo.xpath('lworkcit'):
self.ui.radio_lworkyes.setChecked(True)
self.lworkcit_widget._from_xml(
citeinfo.xpath('lworkcit/citeinfo')[0])
except KeyError:
pass
class Citeinfo(WizardWidget): #
drag_label = "Citation information <citeinfo>"
acceptable_tags = ["citation", "citeinfo"]
def __init__(self, parent=None, include_lwork=True):
self.origin_hint = 'Suggested format "First M. Last"'
self.include_lwork = include_lwork
self.schema = "bdp"
WizardWidget.__init__(self, parent=parent)
self.doi_lookup = None
self.highlighter = Highlighter(self.ui.fgdc_title.document())
self.ui.fgdc_title.textChanged.connect(self.remove_returns)
self.ui.fgdc_title.setMaximumHeight(self.ui.fgdc_geoform.height())
self.ui.fgdc_title.mouseMoveEvent = self.mouse_move
def mouse_move(self, e):
"""
over ride the mouse move event to ignore mouse movement (scrolling),
but only in the vertical dimension. Makes this plain text edit look and
feel like a line edit.
Parameters
----------
e : pyqt event
Returns
-------
None
"""
if (e.y() < self.ui.fgdc_title.height() - 3
and e.x() < self.ui.fgdc_title.width() - 3):
QPlainTextEdit.mouseMoveEvent(self.ui.fgdc_title, e)
self.ui.fgdc_title.verticalScrollBar().setValue(
self.ui.fgdc_title.verticalScrollBar().minimum())
def remove_returns(self):
"""
After changes to the title, we need to make sure that the user has not entered any line returns, since the
title is not intended to be a multiline string. Any line returns "\n" will be replace with a ' '
Returns
-------
None
"""
self.ui.fgdc_title.textChanged.disconnect()
old_position = self.ui.fgdc_title.textCursor().position()
curtext = self.ui.fgdc_title.toPlainText()
newtext = curtext.replace("\n", " ")
self.ui.fgdc_title.setPlainText(newtext)
cursor = self.ui.fgdc_title.textCursor()
cursor.setPosition(old_position)
self.ui.fgdc_title.setTextCursor(cursor)
self.ui.fgdc_title.textChanged.connect(self.remove_returns)
def build_ui(self, ):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_citeinfo.Ui_parent_form()
self.ui.setupUi(self)
if self.include_lwork:
self.lworkcit_widget = Citeinfo(parent=self, include_lwork=False)
self.lworkcit_widget.ui.lbl_dataset_title.setText(
"Larger Work Title")
self.ui.lworkcite_widget.layout().addWidget(self.lworkcit_widget)
self.lworkcit_widget.ui.fgdc_geoform.setEditText("publication")
else:
self.ui.fgdc_lworkcit.hide()
self.include_lworkext_change(self.ui.radio_lworkyes.isChecked())
self.ui.series_ext.hide()
self.ui.pub_ext.hide()
self.ui.pubdate_widget = FGDCDate(
label="YYYYMMDD ",
show_format=False,
required=True,
fgdc_name="fgdc_pubdate",
)
self.ui.pubdate_layout.addWidget(self.ui.pubdate_widget)
self.onlink_list = RepeatingElement(
add_text="Add online link",
remove_text="Remove last",
italic_text=
"Is there a link to the data or the agency that produced it? if so, provide the URL(s) ",
widget_kwargs={
"label": "Link",
"line_name": "fgdc_onlink"
},
)
self.onlink_list.add_another()
self.ui.onlink_layout.addWidget(self.onlink_list)
self.fgdc_origin = RepeatingElement(
add_text="Add originator",
remove_text="Remove last",
italic_text=
"Who created the dataset? List the organization and/or person(s)",
widget_kwargs={
"label": "Originator",
"line_name": "fgdc_origin",
"required": True,
"placeholder_text": self.origin_hint,
"spellings": False,
},
)
self.ui.originator_layout.addWidget(self.fgdc_origin)
self.fgdc_origin.add_another()
self.setup_dragdrop(self)
if not hananero_installed:
self.ui.btn_import_doi.hide()
def connect_events(self):
"""
Connect the appropriate GUI components with the corresponding functions
Returns
-------
None
"""
self.ui.radio_lworkyes.toggled.connect(self.include_lworkext_change)
self.ui.radio_seriesyes.toggled.connect(self.include_seriesext_change)
self.ui.radio_pubinfoyes.toggled.connect(self.include_pubext_change)
self.ui.btn_import_doi.clicked.connect(self.get_doi_citation)
def get_doi_citation(self):
if self.doi_lookup is None:
self.doi_lookup = QDialog(parent=self)
self.doi_lookup_ui = UI_DOICiteinfoImporter.Ui_ImportUsgsUser()
self.doi_lookup_ui.setupUi(self.doi_lookup)
self.doi_lookup_ui.btn_OK.clicked.connect(self.add_doi)
self.doi_lookup_ui.btn_cancel.clicked.connect(self.cancel)
utils.set_window_icon(self.doi_lookup)
self.doi_lookup.show()
def add_doi(self):
doi = self.doi_lookup_ui.le_doi.text()
try:
citeinfo = doi_utils.get_doi_citation(doi)
if citeinfo is None:
msgbox = QMessageBox(self)
utils.set_window_icon(msgbox)
msgbox.setIcon(QMessageBox.Warning)
msg = "'{}' Not Found on DataCite".format(doi)
msg += "\nMake sure the DOI is valid and active."
msgbox.setText(msg)
msgbox.setInformativeText("No matching citation found")
msgbox.setWindowTitle("DOI Not Found")
msgbox.setStandardButtons(QMessageBox.Ok)
msgbox.exec_()
else:
self.from_xml(citeinfo.to_xml())
except:
msg = "We ran into a problem creating a citeinfo element from that DOI({})".format(
doi)
msg += "Check the DOI and/or manually create the citation for it"
QMessageBox.warning(self, "Problem DOI", msg)
self.cancel()
def cancel(self):
self.doi_lookup.deleteLater()
self.doi_lookup = None
def dragEnterEvent(self, e):
"""
Only accept Dragged items that can be converted to an xml object with
a root tag called in our list of acceptable_tags
Parameters
----------
e : qt event
Returns
-------
"""
mime_data = e.mimeData()
if e.mimeData().hasUrls():
if "doi" in e.mimeData().urls()[0].url().lower():
e.accept()
elif e.mimeData().hasFormat("text/plain"):
if self.is_doi_str(mime_data.text()):
e.accept()
else:
try:
element = xml_utils.string_to_node(mime_data.text())
if element is not None and element.tag in self.acceptable_tags:
e.accept()
except AttributeError:
e.ignore()
else:
e.ignore()
def is_doi_str(self, string):
if string.startswith("https://doi.org"):
return True
if string.startswith("doi.org"):
return True
if string.startswith("http://dx.doi.org"):
return True
if string.startswith("doi:"):
return True
cleaned_doi = doi_utils.clean_doi(string).lower().strip()
if cleaned_doi.startswith("doi:"):
return True
return False
def dropEvent(self, e):
"""
Updates the form with the contents of an xml node dropped onto it.
Parameters
----------
e : qt event
Returns
-------
None
"""
try:
e.setDropAction(Qt.CopyAction)
e.accept()
mime_data = e.mimeData()
if mime_data.hasUrls() or self.is_doi_str(mime_data.text()):
if self.is_doi_str(mime_data.text()):
doi = mime_data.text()
else:
doi = e.mimeData().urls()[0].url()
try:
citeinfo = doi_utils.get_doi_citation(doi)
self.from_xml(citeinfo.to_xml())
except:
msg = "We ran into a problem creating a citeinfo element from that DOI({})".format(
doi)
msg += "Check the DOI and/or manually create the citation for it"
QMessageBox.warning(self, "Problem DOI", msg)
else:
element = xml_utils.string_to_node(mime_data.text())
self.from_xml(element)
except:
e = sys.exc_info()[0]
print("problem drop", e)
def include_seriesext_change(self, b):
"""
Extended citation to support series information of the record.
Parameters
----------
b: qt event
Returns
-------
None
"""
if b:
self.ui.series_ext.show()
else:
self.ui.series_ext.hide()
def include_pubext_change(self, b):
"""
Extended citation to support publication information of the record.
Parameters
----------
b: qt event
Returns
-------
None
"""
if b:
self.ui.pub_ext.show()
else:
self.ui.pub_ext.hide()
def include_lworkext_change(self, b):
"""
Extended citation to support a larger body of information for the record.
Parameters
----------
b: qt event
Returns
-------
None
"""
if b:
self.ui.lworkcite_widget.show()
else:
self.ui.lworkcite_widget.hide()
def clear_widget(self):
WizardWidget.clear_widget(self)
self.ui.radio_pubinfono.setChecked(True)
self.ui.radio_seriesno.setChecked(True)
def to_xml(self):
"""
encapsulates the QLineEdit text in an element tag
Returns
-------
citation element tag in xml tree
"""
citeinfo = xml_utils.xml_node("citeinfo")
for origin in self.fgdc_origin.get_widgets():
xml_utils.xml_node("origin",
text=origin.text(),
parent_node=citeinfo)
pubdate = xml_utils.xml_node("pubdate",
text=self.ui.pubdate_widget.get_date(),
parent_node=citeinfo)
title = xml_utils.xml_node("title",
self.ui.fgdc_title.toPlainText(),
parent_node=citeinfo)
if self.ui.fgdc_edition.text():
othercit = xml_utils.xml_node("edition",
self.ui.fgdc_edition.text(),
parent_node=citeinfo)
geoform = xml_utils.xml_node("geoform",
self.ui.fgdc_geoform.currentText(),
parent_node=citeinfo)
if self.ui.radio_seriesyes.isChecked():
serinfo = xml_utils.xml_node("serinfo", parent_node=citeinfo)
sername = xml_utils.xml_node("sername",
text=self.ui.fgdc_sername.text(),
parent_node=serinfo)
issue = xml_utils.xml_node("issue",
text=self.ui.fgdc_issue.text(),
parent_node=serinfo)
if self.ui.radio_pubinfoyes.isChecked() and (
self.ui.fgdc_pubplace.text() != ""
or self.ui.fgdc_publish.text() != ""):
pubinfo = xml_utils.xml_node("pubinfo", parent_node=citeinfo)
pubplace = xml_utils.xml_node("pubplace",
parent_node=pubinfo,
text=self.ui.fgdc_pubplace.text())
publish = xml_utils.xml_node("publish",
parent_node=pubinfo,
text=self.ui.fgdc_publish.text())
if self.ui.fgdc_othercit.toPlainText():
othercit = xml_utils.xml_node("othercit",
self.ui.fgdc_othercit.toPlainText(),
parent_node=citeinfo)
for onlink in self.onlink_list.get_widgets():
if onlink.text() != "":
onlink_node = xml_utils.xml_node("onlink",
parent_node=citeinfo,
text=onlink.text())
if self.include_lwork and self.ui.radio_lworkyes.isChecked():
lworkcit = xml_utils.xml_node("lworkcit", parent_node=citeinfo)
lwork = self.lworkcit_widget.to_xml()
lworkcit.append(lwork)
return citeinfo
def from_xml(self, citeinfo):
"""
parses the xml code into the relevant citation elements
Parameters
----------
citation - the xml element status and its contents
Returns
-------
None
"""
self.original_xml = citeinfo
self.clear_widget()
try:
if citeinfo.tag == "citation":
citeinfo = citeinfo.xpath("citeinfo")[0]
elif citeinfo.tag != "citeinfo":
print("The tag is not 'citation' or 'citeinfo'")
return
self.fgdc_origin.clear_widgets(add_another=False)
originators = citeinfo.findall("origin")
if originators:
self.fgdc_origin.clear_widgets(add_another=False)
for origin in citeinfo.findall("origin"):
origin_widget = self.fgdc_origin.add_another()
origin_widget.setText(origin.text)
else:
self.fgdc_origin.add_another()
if citeinfo.findall("geoform"):
self.ui.fgdc_geoform.setEditText(
citeinfo.findall("geoform")[0].text)
utils.populate_widget_element(
self.ui.pubdate_widget.ui.fgdc_caldate, citeinfo, "pubdate")
utils.populate_widget_element(self.ui.fgdc_title, citeinfo,
"title")
utils.populate_widget_element(self.ui.fgdc_edition, citeinfo,
"edition")
utils.populate_widget_element(self.ui.fgdc_othercit, citeinfo,
"othercit")
self.onlink_list.clear_widgets()
if citeinfo.findall("onlink"):
self.onlink_list.clear_widgets(add_another=False)
for onlink in citeinfo.findall("onlink"):
self.onlink_list.add_another()
onlink_widget = self.onlink_list.widgets[-1]
onlink_widget.setText(onlink.text)
if citeinfo.xpath("serinfo"):
self.ui.radio_seriesyes.setChecked(True)
utils.populate_widget(self, citeinfo.xpath("serinfo")[0])
else:
self.ui.radio_seriesyes.setChecked(False)
pubinfo = citeinfo.xpath("pubinfo")
if pubinfo:
self.ui.radio_pubinfoyes.setChecked(True)
utils.populate_widget_element(self.ui.fgdc_publish, pubinfo[0],
"publish")
utils.populate_widget_element(self.ui.fgdc_pubplace,
pubinfo[0], "pubplace")
else:
self.ui.radio_pubinfoyes.setChecked(False)
self.ui.radio_pubinfono.setChecked(True)
if citeinfo.xpath("lworkcit"):
try:
self.ui.radio_lworkyes.setChecked(True)
self.lworkcit_widget.from_xml(
citeinfo.xpath("lworkcit/citeinfo")[0])
except AttributeError:
msg = "you pasted a citation element into the larger work citation area"
msg += "\n that contained a larger work citation"
msg += "\n Multiple nested larger work citations are not currently supported in the tool"
msg += "\n\n the larger work citation being pasted will be ignored"
QMessageBox.warning(self, "Dropped Content Warning", msg)
else:
self.ui.radio_lworkno.setChecked(True)
except KeyError:
pass
class Timeperd(WizardWidget): #
drag_label = "Time Period of Content <timeperd>"
acceptable_tags = ['timeperd']
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_timeperd.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.single_date = FGDCDate(label=' Single Date ',
fgdc_name='fgdc_caldate')
self.ui.fgdc_sngdate.layout().insertWidget(0, self.single_date)
self.range_start_date = FGDCDate(label='Start ',
fgdc_name='fgdc_begdate')
self.range_end_date = FGDCDate(label='End ', fgdc_name='fgdc_enddate')
self.ui.layout_daterange.addWidget(self.range_start_date)
self.ui.layout_daterange.addWidget(self.range_end_date)
date_widget_kwargs = {
'show_format': False,
'label': 'Individual Date ',
'fgdc_name': 'fgdc_caldate',
'parent_fgdc_name': 'fgdc_sngdate'
}
self.multi_dates = RepeatingElement(widget=FGDCDate,
widget_kwargs=date_widget_kwargs)
self.multi_dates.add_another()
self.switch_primary()
def connect_events(self):
"""
Connect the appropriate GUI components with the corresponding functions
Returns
-------
None
"""
self.ui.radio_single.toggled.connect(self.switch_primary)
self.ui.radio_range.toggled.connect(self.switch_primary)
self.ui.radio_multiple.toggled.connect(self.switch_primary)
def switch_primary(self):
"""
Switches form to reflect either organization or person primary
Returns
-------
None
"""
if self.ui.radio_single.isChecked():
self.findChild(QStackedWidget, "fgdc_timeinfo").setCurrentIndex(0)
self.ui.fgdc_sngdate.show()
self.ui.fgdc_rngdates.hide()
self.ui.fgdc_mdattim.hide()
self.ui.fgdc_mdattim.layout().removeWidget(self.multi_dates)
elif self.ui.radio_range.isChecked():
self.findChild(QStackedWidget, "fgdc_timeinfo").setCurrentIndex(1)
self.ui.fgdc_rngdates.hide()
self.ui.fgdc_rngdates.show()
self.ui.fgdc_mdattim.hide()
self.ui.fgdc_mdattim.layout().removeWidget(self.multi_dates)
elif self.ui.radio_multiple.isChecked():
self.findChild(QStackedWidget, "fgdc_timeinfo").setCurrentIndex(2)
self.ui.fgdc_sngdate.hide()
self.ui.fgdc_rngdates.hide()
self.ui.fgdc_mdattim.layout().addWidget(self.multi_dates)
self.ui.fgdc_mdattim.show()
def _to_xml(self):
"""
encapsulates the QTabWidget text for Metadata Time in an element tag
Returns
-------
timeperd element tag in xml tree
"""
timeperd = xml_utils.xml_node('timeperd')
timeinfo = xml_utils.xml_node("timeinfo", parent_node=timeperd)
tabIndex = self.ui.fgdc_timeinfo.currentIndex()
if tabIndex == 0:
sngdate = xml_utils.xml_node("sngdate", parent_node=timeinfo)
caldate = xml_utils.xml_node('caldate',
parent_node=sngdate,
text=self.single_date.get_date())
if tabIndex == 1:
rngdates = xml_utils.xml_node("rngdates", parent_node=timeinfo)
begdate = xml_utils.xml_node("begdate",
parent_node=rngdates,
text=self.range_start_date.get_date())
enddate = xml_utils.xml_node("enddate",
parent_node=rngdates,
text=self.range_end_date.get_date())
if tabIndex == 2:
mdattim = xml_utils.xml_node("mdattim", parent_node=timeinfo)
for single_date in self.multi_dates.get_widgets():
single_date_node = xml_utils.xml_node("sngdate",
parent_node=mdattim)
caldate = xml_utils.xml_node('caldate',
parent_node=single_date_node,
text=single_date.get_date())
current = xml_utils.xml_node('current',
parent_node=timeperd,
text=self.ui.fgdc_current.currentText())
return timeperd
def _from_xml(self, timeperd):
"""
parses the xml code into the relevant timeperd elements
Parameters
----------
metadata_date - the xml element timeperd and its contents
Returns
-------
None
"""
try:
if timeperd.tag == 'timeperd':
if timeperd.findall("current"):
current_text = timeperd.findtext("current")
current_box = self.findChild(QComboBox, 'fgdc_current')
current_box.setCurrentText(current_text)
else:
pass
timeinfo_stack = self.ui.fgdc_timeinfo
if timeperd.xpath("timeinfo/rngdates"):
self.ui.radio_range.setChecked(True)
timeinfo_stack.setCurrentIndex(1)
begdate = timeperd.findtext("timeinfo/rngdates/begdate")
self.range_start_date.set_date(begdate)
enddate = timeperd.findtext("timeinfo/rngdates/enddate")
self.range_end_date.set_date(enddate)
elif timeperd.xpath("timeinfo/mdattim"):
self.ui.radio_multiple.setChecked(True)
timeinfo_stack.setCurrentIndex(2)
self.multi_dates.clear_widgets(add_another=False)
for caldate in timeperd.xpath(
'timeinfo/mdattim/sngdate/caldate'):
date_widget = self.multi_dates.add_another()
date_widget.set_date(caldate.text)
elif timeperd.xpath("timeinfo/sngdate"):
self.ui.radio_single.setChecked(True)
timeinfo_stack.setCurrentIndex(0)
sngdate = timeperd.findtext("timeinfo/sngdate/caldate")
self.single_date.set_date(sngdate)
else:
pass
else:
print("The tag is not timeperd")
except KeyError:
pass
class ThemeList(WizardWidget): #
drag_label = "Theme Keywords <keywords>"
acceptable_tags = ['keywords']
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_theme_list.Ui_theme_list()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.ui.theme_tabs.setStyleSheet("QTabBar::tab::disabled {width: 0; height: 0; margin: 0; padding: 0; border: none;} ")
self.iso_kws = RepeatingElement(which='vertical',
add_text='Add additonal',
widget=IsoKeyword,
remove_text='Remove Keyword',
italic_text='ISO Topic Category Keywords')
self.iso_kws.add_another()
self.ui.iso_keywords_layout.addWidget(self.iso_kws)
self.thesauri = []
def connect_events(self):
"""
Connect the appropriate GUI components with the corresponding functions
Returns
-------
None
"""
self.ui.btn_add_thesaurus.clicked.connect(self.add_another)
self.ui.btn_remove_selected.clicked.connect(self.remove_selected)
self.ui.btn_add_iso.clicked.connect(self.add_iso)
self.ui.btn_search_controlled.clicked.connect(self.search_controlled)
def add_another(self, click=False, tab_label='', locked=False):
if 'None' not in [t.get_thesaurus_name() for t in self.thesauri] and \
tab_label == '':
theme_widget = self.add_keyword(keyword='', thesaurus='None',
locked=False)
else:
theme_widget = Theme()
theme_widget.ui.fgdc_themekt.textChanged.connect(self.changed_thesaurus)
self.ui.theme_tabs.addTab(theme_widget, tab_label)
self.ui.theme_tabs.setCurrentIndex(self.ui.theme_tabs.count()-1)
self.thesauri.append(theme_widget)
return theme_widget
def changed_thesaurus(self, s):
current_index = self.ui.theme_tabs.currentIndex()
current_tab = self.ui.theme_tabs.setTabText(current_index, 'Thesaurus: ' + s)
def remove_selected(self):
current_index = self.ui.theme_tabs.currentIndex()
if current_index == 0:
self.remove_iso()
else:
self.ui.theme_tabs.removeTab(current_index)
del self.thesauri[current_index-1]
def remove_iso(self):
self.ui.theme_tabs.setTabEnabled(0, False)
self.ui.fgdc_theme.hide()
def add_iso(self):
self.ui.theme_tabs.setTabEnabled(0, True)
self.ui.fgdc_theme.show()
self.ui.theme_tabs.setCurrentIndex(0)
self.repaint()
def get_children(self, widget):
children = []
if self.ui.theme_tabs.isTabEnabled(0):
children.append(self.ui.fgdc_theme)
for theme in self.thesauri:
children.append(theme)
return children
def clear_widget(self, remove_iso=False):
self.iso_kws.clear_widgets()
if remove_iso:
self.remove_iso()
for i in range(len(self.thesauri), 0, -1):
self.ui.theme_tabs.setCurrentIndex(i)
self.remove_selected()
def search_controlled(self):
self.thesaurus_search = ThesaurusSearch.ThesaurusSearch(add_term_function=self.add_keyword, parent=self)
self.thesaurus_search.setWindowTitle('Search USGS Controlled Vocabularies')
self.thesaurus_search.show()
def add_keyword(self, keyword=None, thesaurus=None, locked=True):
theme_widget = None
for theme in self.thesauri:
if theme.ui.fgdc_themekt.text() == thesaurus:
theme_widget = theme
if theme_widget is None:
shortname = thesaurus.split(' ')[0]
theme_widget = self.add_another(tab_label=shortname, locked=locked)
theme_widget.ui.fgdc_themekt.setText(thesaurus)
if locked:
theme_widget.lock()
self.changed_thesaurus(shortname)
theme_widget.add_keyword(keyword, locked=locked)
return theme_widget
def _to_xml(self):
"""
encapsulates the QPlainTextEdit text in an element tag
Returns
-------
procstep element tag in xml tree
"""
keywords = xml_utils.xml_node('keywords')
if self.ui.theme_tabs.isTabEnabled(0):
theme = xml_utils.xml_node('theme', parent_node=keywords)
themekt = xml_utils.xml_node('themekt',
text=self.ui.fgdc_themekt.text(),
parent_node=theme)
for isokw in self.iso_kws.get_widgets():
themekey = xml_utils.xml_node('themekey', text=isokw.ui.fgdc_themekey.currentText(),
parent_node=theme)
for theme in self.thesauri:
theme_xml = theme._to_xml()
keywords.append(theme_xml)
return keywords
def _from_xml(self, keywords_xml):
"""
parses the xml code into the relevant procstep elements
Parameters
----------
process_step - the xml element status and its contents
Returns
-------
None
"""
self.clear_widget(remove_iso=True)
self.original_xml = keywords_xml
if keywords_xml.tag == 'keywords':
for theme_xml in xml_utils.search_xpath(keywords_xml, 'theme', False):
themekt = xml_utils.get_text_content(theme_xml, 'themekt')
if themekt is not None and 'iso 19115' in themekt.lower():
self.add_iso()
self.iso_kws.clear_widgets(add_another=False)
for themekey in xml_utils.search_xpath(theme_xml,
'themekey',
only_first=False):
iso = self.iso_kws.add_another()
iso.ui.fgdc_themekey.setCurrentText(themekey.text)
else:
theme = self.add_another()
theme._from_xml(theme_xml)
class KeywordsRepeater(WizardWidget):
drag_label = "NA <NA>"
def __init__(
self,
thesaurus_label="Thesaurus",
keywords_label="Keywords:",
line_name="kw",
spellings=True,
parent=None,
):
self.thesaurus_label = thesaurus_label
self.keywords_label = keywords_label
self.line_name = line_name
self.keywords = None
self.spellings = spellings
WizardWidget.__init__(self, parent=parent)
def build_ui(self):
"""
Build and modify this widget's GUI
Returns
-------
None
"""
self.ui = UI_keywords_repeater.Ui_Form()
self.ui.setupUi(self)
self.setup_dragdrop(self)
self.ui.thesaurus_label = self.thesaurus_label
widget_kwargs = {
"label": self.keywords_label,
"line_name": self.line_name,
"required": True,
"spellings": self.spellings,
}
self.keywords = RepeatingElement(
add_text="Add keyword",
remove_text="Remove last",
widget_kwargs=widget_kwargs,
)
self.keywords.ui.italic_label.setStyleSheet("")
self.keywords.add_another()
self.ui.keywords_layout.insertWidget(0, self.keywords)
def lock(self):
self.ui.fgdc_themekt.setReadOnly(True)
self.keywords.ui.addAnother.setEnabled(False)
def get_keywords(self):
return [kw.text() for kw in self.keywords.get_widgets()]
def add_another(self, locked=False):
widget = self.keywords.add_another()
widget.setObjectName(self.line_name)
widget.added_line.setReadOnly(locked)
return widget
def get_widgets(self):
return self.keywords.get_widgets()
