importhub stub
This commit is contained in:
Laura Klünder
parent
fadbabf9bf
commit
4d97af63b3
4 changed files with 18 additions and 200 deletions
|
|
@ -1,199 +0,0 @@
|
|||
import argparse
|
||||
import logging
|
||||
import re
|
||||
from xml.etree import ElementTree
|
||||
|
||||
from django.core.management.base import BaseCommand, CommandError
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
from shapely.affinity import scale, translate
|
||||
from shapely.geometry import Polygon
|
||||
|
||||
from c3nav.mapdata.models import Area, MapUpdate, Obstacle, Space
|
||||
from c3nav.mapdata.utils.cache.changes import changed_geometries
|
||||
|
||||
|
||||
class Command(BaseCommand):
|
||||
help = 'render the map'
|
||||
|
||||
@staticmethod
|
||||
def space_value(value):
|
||||
try:
|
||||
space = Space.objects.get(pk=value)
|
||||
except Space.DoesNotExist:
|
||||
raise argparse.ArgumentTypeError(
|
||||
_('unknown space')
|
||||
)
|
||||
return space
|
||||
|
||||
def add_arguments(self, parser):
|
||||
parser.add_argument('svgfile', type=argparse.FileType('r'), help=_('svg file to import'))
|
||||
parser.add_argument('name', type=str, help=_('name of the import'))
|
||||
parser.add_argument('--type', type=str, required=True, choices=('buildings', 'areas', 'obstacles'),
|
||||
help=_('type of objects to create'))
|
||||
parser.add_argument('--space', type=self.space_value, required=True,
|
||||
help=_('space to add the objects to'))
|
||||
parser.add_argument('--minx', type=float, required=True,
|
||||
help=_('minimum x coordinate, everthing left of it will be cropped'))
|
||||
parser.add_argument('--miny', type=float, required=True,
|
||||
help=_('minimum y coordinate, everthing below it will be cropped'))
|
||||
parser.add_argument('--maxx', type=float, required=True,
|
||||
help=_('maximum x coordinate, everthing right of it will be cropped'))
|
||||
parser.add_argument('--maxy', type=float, required=True,
|
||||
help=_('maximum y coordinate, everthing above it will be cropped'))
|
||||
|
||||
@staticmethod
|
||||
def parse_svg_data(data):
|
||||
first = False
|
||||
|
||||
last_point = (0, 0)
|
||||
last_end_point = None
|
||||
|
||||
done_subpaths = []
|
||||
current_subpath = []
|
||||
while data:
|
||||
data = data.lstrip().replace(',', ' ')
|
||||
command = data[0]
|
||||
if first and command not in 'Mm':
|
||||
raise ValueError('path data has to start with moveto command.')
|
||||
data = data[1:].lstrip()
|
||||
first = False
|
||||
|
||||
numbers = []
|
||||
while True:
|
||||
match = re.match(r'^-?[0-9]+(\.[0-9]+)?(e-?[0-9]+)?', data)
|
||||
if match is None:
|
||||
break
|
||||
numbers.append(float(match.group(0)))
|
||||
data = data[len(match.group(0)):].lstrip()
|
||||
|
||||
relative = command.islower()
|
||||
if command in 'Mm':
|
||||
if not len(numbers) or len(numbers) % 2:
|
||||
raise ValueError('Invalid number of arguments for moveto command!')
|
||||
numbers = iter(numbers)
|
||||
first = True
|
||||
for x, y in zip(numbers, numbers):
|
||||
if relative:
|
||||
x, y = last_point[0] + x, last_point[1] + y
|
||||
if first:
|
||||
first = False
|
||||
if current_subpath:
|
||||
done_subpaths.append(current_subpath)
|
||||
last_end_point = current_subpath[-1]
|
||||
current_subpath = []
|
||||
current_subpath.append((x, y))
|
||||
last_point = (x, y)
|
||||
|
||||
elif command in 'Ll':
|
||||
if not len(numbers) or len(numbers) % 2:
|
||||
raise ValueError('Invalid number of arguments for lineto command!')
|
||||
numbers = iter(numbers)
|
||||
for x, y in zip(numbers, numbers):
|
||||
if relative:
|
||||
x, y = last_point[0] + x, last_point[1] + y
|
||||
if not current_subpath:
|
||||
current_subpath.append(last_end_point)
|
||||
current_subpath.append((x, y))
|
||||
last_point = (x, y)
|
||||
|
||||
elif command in 'Hh':
|
||||
if not len(numbers):
|
||||
raise ValueError('Invalid number of arguments for horizontal lineto command!')
|
||||
y = last_point[1]
|
||||
for x in numbers:
|
||||
if relative:
|
||||
x = last_point[0] + x
|
||||
if not current_subpath:
|
||||
current_subpath.append(last_end_point)
|
||||
current_subpath.append((x, y))
|
||||
last_point = (x, y)
|
||||
|
||||
elif command in 'Vv':
|
||||
if not len(numbers):
|
||||
raise ValueError('Invalid number of arguments for vertical lineto command!')
|
||||
x = last_point[0]
|
||||
for y in numbers:
|
||||
if relative:
|
||||
y = last_point[1] + y
|
||||
if not current_subpath:
|
||||
current_subpath.append(last_end_point)
|
||||
current_subpath.append((x, y))
|
||||
last_point = (x, y)
|
||||
|
||||
elif command in 'Zz':
|
||||
if numbers:
|
||||
raise ValueError('Invalid number of arguments for closepath command!')
|
||||
current_subpath.append(current_subpath[0])
|
||||
done_subpaths.append(current_subpath)
|
||||
last_end_point = current_subpath[-1]
|
||||
current_subpath = []
|
||||
|
||||
else:
|
||||
raise ValueError('unknown svg command: ' + command)
|
||||
|
||||
if current_subpath:
|
||||
done_subpaths.append(current_subpath)
|
||||
return done_subpaths
|
||||
|
||||
def handle(self, *args, **options):
|
||||
minx = options['minx']
|
||||
miny = options['miny']
|
||||
maxx = options['maxx']
|
||||
maxy = options['maxy']
|
||||
|
||||
if minx >= maxx:
|
||||
raise CommandError(_('minx has to be lower than maxx'))
|
||||
if miny >= maxy:
|
||||
raise CommandError(_('miny has to be lower than maxy'))
|
||||
|
||||
width = maxx-minx
|
||||
height = maxy-miny
|
||||
|
||||
model = {'areas': Area, 'obstacles': Obstacle}[options['type']]
|
||||
|
||||
namespaces = {'svg': 'http://www.w3.org/2000/svg'}
|
||||
|
||||
svg = ElementTree.fromstring(options['svgfile'].read())
|
||||
svg_width = float(svg.attrib['width'])
|
||||
svg_height = float(svg.attrib['height'])
|
||||
svg_viewbox = svg.attrib.get('viewBox')
|
||||
|
||||
if svg_viewbox:
|
||||
offset_x, offset_y, svg_width, svg_height = [float(i) for i in svg_viewbox.split(' ')]
|
||||
else:
|
||||
offset_x, offset_y = 0, 0
|
||||
|
||||
for element in svg.findall('.//svg:clipPath/..', namespaces):
|
||||
for clippath in element.findall('./svg:clipPath', namespaces):
|
||||
element.remove(clippath)
|
||||
|
||||
for element in svg.findall('.//svg:symbol/..', namespaces):
|
||||
for clippath in element.findall('./svg:symbol', namespaces):
|
||||
element.remove(clippath)
|
||||
|
||||
if svg.findall('.//*[@transform]'):
|
||||
raise CommandError(_('svg contains transform attributes. Use inkscape apply transforms.'))
|
||||
|
||||
if model.objects.filter(space=options['space'], import_tag=options['name']).exists():
|
||||
raise CommandError(_('objects with this import tag already exist in this space.'))
|
||||
|
||||
with MapUpdate.lock():
|
||||
changed_geometries.reset()
|
||||
for path in svg.findall('.//svg:path', namespaces):
|
||||
for polygon in self.parse_svg_data(path.attrib['d']):
|
||||
if len(polygon) < 3:
|
||||
continue
|
||||
polygon = Polygon(polygon)
|
||||
polygon = translate(polygon, xoff=-offset_x, yoff=-offset_y)
|
||||
polygon = scale(polygon, xfact=1, yfact=-1, origin=(0, svg_height/2))
|
||||
polygon = scale(polygon, xfact=width / svg_width, yfact=height / svg_height, origin=(0, 0))
|
||||
polygon = translate(polygon, xoff=minx, yoff=miny)
|
||||
obj = model(geometry=polygon, space=options['space'], import_tag=options['name'])
|
||||
obj.save()
|
||||
MapUpdate.objects.create(type='importsvg')
|
||||
|
||||
logger = logging.getLogger('c3nav')
|
||||
logger.info('Imported, map update created.')
|
||||
logger.info('Next step: go into the shell and edit them using '
|
||||
'%s.objects.filter(space_id=%r, import_tag=%r)' %
|
||||
(model.__name__, options['space'].pk, options['name']))
|
||||
14
src/c3nav/mapdata/management/commands/importhub.py
Normal file
14
src/c3nav/mapdata/management/commands/importhub.py
Normal file
|
|
@ -0,0 +1,14 @@
|
|||
import requests
|
||||
from django.conf import settings
|
||||
from django.core.management.base import BaseCommand
|
||||
|
||||
|
||||
class Command(BaseCommand):
|
||||
help = 'import from hub'
|
||||
|
||||
def handle(self, *args, **options):
|
||||
r = requests.get(settings.HUB_API_BASE+"/integration/c3nav",
|
||||
headers={"Authorization": "Token "+settings.HUB_API_SECRET})
|
||||
r.raise_for_status()
|
||||
from pprint import pprint
|
||||
pprint(r.json())
|
||||
|
|
@ -13,7 +13,7 @@ from c3nav.mapdata.utils.cache.changes import changed_geometries
|
|||
|
||||
|
||||
class Command(BaseCommand):
|
||||
help = 'render the map'
|
||||
help = 'import svg file'
|
||||
|
||||
@staticmethod
|
||||
def space_value(value):
|
||||
|
|
|
|||
|
|
@ -176,6 +176,9 @@ if len(INITIAL_BOUNDS) == 4:
|
|||
else:
|
||||
INITIAL_BOUNDS = None
|
||||
|
||||
HUB_API_BASE = config.get('c3nav', 'hub_api_base', fallback='').removesuffix('/')
|
||||
HUB_API_SECRET = config.get('c3nav', 'hub_api_secret', fallback='')
|
||||
|
||||
_db_backend = config.get('database', 'backend', fallback='sqlite3')
|
||||
DATABASES: dict[str, dict[str, str | int | Path]] = {
|
||||
'default': env.db_url('C3NAV_DATABASE') if 'C3NAV_DATABASE' in env else {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue