def import_data(project: 'Project', class_: str, data: list[Any]) -> None:
from openatlas.models.entity import Entity
from openatlas.models.gis import Gis
for row in data:
entity = Entity.insert(
class_, row['name'],
row['description'] if 'description' in row else None)
Db.import_data(
project.id,
entity.id,
current_user.id,
origin_id=row['id'] if 'id' in row and row['id'] else None)
# Dates
dates = {
'begin_from': None,
'begin_to': None,
'begin_comment': None,
'end_from': None,
'end_to': None,
'end_comment': None
}
if 'begin_from' in row and row['begin_from']:
dates['begin_from'] = row['begin_from']
if 'begin_to' in row and row['begin_to']:
dates['begin_to'] = row['begin_to']
if 'begin_comment' in row and row['begin_comment']:
dates['begin_comment'] = row['begin_comment']
if 'end_from' in row and row['end_from']:
dates['end_from'] = row['end_from']
if 'end_to' in row and row['end_to']:
dates['end_to'] = row['end_to']
if 'end_comment' in row and row['end_comment']:
dates['end_comment'] = row['end_comment']
entity.update({'attributes': dates})
# Types
if 'type_ids' in row and row['type_ids']: # pragma: no cover
for type_id in str(row['type_ids']).split():
if not Import.check_type_id(type_id, class_):
continue
entity.link('P2', g.types[int(type_id)])
# GIS
if class_ == 'place':
location = Entity.insert('object_location',
f"Location of {row['name']}")
entity.link('P53', location)
if 'easting' in row \
and is_float(row['easting']) \
and 'northing' in row \
and is_float(row['northing']):
Gis.insert_import(entity=entity,
location=location,
project=project,
easting=row['easting'],
northing=row['northing'])
def update_project(project: Project) -> None:
from openatlas.util.util import sanitize
Db.update_project(project.id, project.name,
sanitize(project.description, 'text'))
def check_duplicates(class_: str, names: List[str]) -> List[str]:
return Db.check_duplicates(class_, names)
def get_origin_ids(project: Project, origin_ids: List[str]) -> List[str]:
return Db.check_origin_ids(project.id, origin_ids)
def delete_project(id_: int) -> None:
Db.delete_project(id_)
def get_project_by_name(name: str) -> Optional[Project]:
row = Db.get_project_by_name(name)
return Project(row) if row else None
def get_project_by_id(id_: int) -> Project:
return Project(Db.get_project_by_id(id_))
def get_all_projects() -> List[Project]:
return [Project(row) for row in Db.get_all_projects()]
def insert_project(name: str, description: Optional[str] = None) -> int:
return Db.insert_project(name,
description.strip() if description else None)
def update_project(project: Project) -> None:
Db.update_project(project.id, project.name,
sanitize(project.description, 'text'))
