def load_csv_file_to_layer(
layer,
organization_slug,
project_slug,
attribute):
"""Check if csv file is exist for layer,
then add attribute from csv file to layer
:param layer: layer to be added
:type layer: QgsVectorLayer
:param organization_slug: organization slug name
:type organization_slug: str
:param project_slug: project slug name
:type project_slug: str
:param attribute: additional csv attribute name
:type attribute: str
:return:
"""
file_path = get_csv_path(organization_slug, project_slug, attribute)
if os.path.isfile(file_path):
with open(file_path, 'rb') as csv_file:
reader = csv.reader(csv_file, delimiter=',', quotechar='"')
next(reader, None)
layer.startEditing()
for row in reader:
feature = QgsFeature()
feature.setAttributes(row)
layer.addFeature(feature, True)
layer.commitChanges()
def add_tabular_layer(
tabular_layer,
organization_slug,
project_slug,
attribute):
"""Add a tabular layer to the folder.
:param tabular_layer: The layer to add.
:type tabular_layer: QgsVectorLayer
:param organization_slug: organization slug for data
:type organization_slug: str
:param project_slug: project_slug for getting spatial
:type project_slug: str
:param attribute: additional csv name
:type attribute: str
:returns: A two-tuple. The first element will be True if we could add
the layer to the datastore. The second element will be the layer
name which has been used or the error message.
:rtype: (bool, str)
"""
file_path = get_csv_path(organization_slug, project_slug, attribute)
QgsVectorFileWriter.writeAsVectorFormat(
tabular_layer,
file_path,
'utf-8',
None,
'CSV')
return True, file_path
def relationships_layer(self, vector_layers):
"""Create relationship layer.
:param vector_layers: List of QGS vector layer in memory
:type vector_layers: list of QgsVectorLayer
:return: Relationship layer
:rtype: QgsVectorLayer
"""
organization_slug = self.project['organization']['slug']
project_slug = self.project['slug']
attribute = 'relationships'
api = u'/api/v1/organizations/{organization_slug}/projects/' \
u'{project_slug}/spatial/{spatial_unit_id}/relationships/'
csv_path = get_csv_path(organization_slug, project_slug, attribute)
if os.path.isfile(csv_path):
os.remove(csv_path)
relationship_layer = tools.create_memory_layer(
layer_name='%s/%s/%s' %
(organization_slug, project_slug, attribute),
geometry=QGis.NoGeometry,
fields=[
QgsField('spatial_id', QVariant.String, "string"),
QgsField('rel_id', QVariant.String, "string"),
QgsField('rel_name', QVariant.String, "string"),
QgsField('party_id', QVariant.String, "string"),
QgsField('attributes', QVariant.String, "string"),
])
QgsMapLayerRegistry.instance().addMapLayer(relationship_layer)
for vector_layer in vector_layers:
# Add relationship layer id to spatial attribute table
spatial_id_index = vector_layer.fieldNameIndex('id')
for index, feature in enumerate(vector_layer.getFeatures()):
attributes = feature.attributes()
spatial_unit_id = attributes[spatial_id_index]
spatial_api = api.format(organization_slug=organization_slug,
project_slug=project_slug,
spatial_unit_id=spatial_unit_id)
connector = ApiConnect(get_url_instance() + spatial_api)
status, results = connector.get(paginated=True)
if not status or len(results) == 0:
continue
try:
for result in results:
relationship_layer.startEditing()
fet = QgsFeature()
questionnaire_attr = result['attributes']
if not questionnaire_attr:
questionnaire_attr = '-'
else:
questionnaire_attr = json.dumps(questionnaire_attr)
fet.setAttributes([
attributes[spatial_id_index],
result['id'],
result['tenure_type'],
result['party']['id'],
questionnaire_attr,
])
relationship_layer.addFeature(fet, True)
relationship_layer.commitChanges()
except (IndexError, KeyError):
continue
self.process_attributes(relationship_layer)
Utilities.add_tabular_layer(relationship_layer, organization_slug,
project_slug, attribute)
return relationship_layer
def parties_layer(self):
"""Create parties layer.
:param vector_layer: QGS vector layer in memory
:type vector_layer: QgsVectorLayer
"""
organization_slug = self.project['organization']['slug']
project_slug = self.project['slug']
attribute = 'parties'
csv_path = get_csv_path(organization_slug, project_slug, attribute)
if os.path.isfile(csv_path):
os.remove(csv_path)
api = u'/api/v1/organizations/{organization_slug}/projects/' \
u'{project_slug}/parties/'.format(
organization_slug=organization_slug,
project_slug=project_slug)
connector = ApiConnect(get_url_instance() + api)
status, results = connector.get(paginated=True)
if not status:
return
party_layer = tools.create_memory_layer(
layer_name='%s/%s/%s' %
(organization_slug, project_slug, attribute),
geometry=QGis.NoGeometry,
fields=[
QgsField('id', QVariant.String, "string"),
QgsField('name', QVariant.String, "string"),
QgsField('type', QVariant.String, "string"),
QgsField('attributes', QVariant.String, "string"),
])
QgsMapLayerRegistry.instance().addMapLayer(party_layer)
for party in results:
party_layer.startEditing()
feature = QgsFeature()
questionnaire_attr = party['attributes']
if not questionnaire_attr:
questionnaire_attr = '-'
else:
questionnaire_attr = json.dumps(questionnaire_attr)
feature.setAttributes([
party['id'], party['name'], party['type'], questionnaire_attr
])
party_layer.addFeature(feature, True)
party_layer.commitChanges()
QCoreApplication.processEvents()
self.process_attributes(party_layer)
Utilities.add_tabular_layer(party_layer, organization_slug,
project_slug, attribute)
return party_layer