Integrate the new Location class

- the old `UTMCoordinate` class becomes the new `Location` class
- its main purpose is to represent locations in both lat-long
  coordinates as well as in the UTM system
- remove `Address.__init__()` and `City.__init__()` methods as they
  are not executed for entries retrieved from the database
- simplfiy the `Location.__init__()` => remove `relative_to` argument
This commit is contained in:
Alexander Hess 2021-01-04 20:33:10 +01:00
commit a1cbb808fd
Signed by: alexander
GPG key ID: 344EA5AB10D868E0
8 changed files with 198 additions and 157 deletions

View file

@ -1,6 +1,5 @@
"""Provide the ORM's `Address` model."""
from typing import Any
import sqlalchemy as sa
from sqlalchemy import orm
@ -69,14 +68,9 @@ class Address(meta.Base):
)
pixels = orm.relationship('AddressPixelAssociation', back_populates='address')
def __init__(self, *args: Any, **kwargs: Any) -> None:
"""Create a new address."""
# Call SQLAlchemy's default `.__init__()` method.
super().__init__(*args, **kwargs)
self._utm_coordinates = utils.UTMCoordinate(
self.latitude, self.longitude, relative_to=self.city.as_origin,
)
# We do not implement a `.__init__()` method and leave that to SQLAlchemy.
# Instead, we use `hasattr()` to check for uninitialized attributes.
# grep:b1f68d24 pylint:disable=attribute-defined-outside-init
def __repr__(self) -> str:
"""Non-literal text representation."""
@ -95,12 +89,39 @@ class Address(meta.Base):
"""
return self.id == self._primary_id
@property
def location(self) -> utils.Location:
"""The location of the address.
The returned `Location` object relates to `.city.viewport.southwest`.
See also the `.x` and `.y` properties that are shortcuts for
`.location.x` and `.location.y`.
Implementation detail: This property is cached as none of the
underlying attributes to calculate the value are to be changed.
"""
if not hasattr(self, '_location'): # noqa:WPS421 note:b1f68d24
self._location = utils.Location(self.latitude, self.longitude)
self._location.relate_to(self.city.as_xy_origin)
return self._location
@property
def x(self) -> int: # noqa=WPS111
"""The `.easting` of the address in meters, relative to the `.city`."""
return self._utm_coordinates.x
"""The relative x-coordinate within the `.city` in meters.
On the implied x-y plane, the `.city`'s southwest corner is the origin.
Shortcut for `.location.x`.
"""
return self.location.x
@property
def y(self) -> int: # noqa=WPS111
"""The `.northing` of the address in meters, relative to the `.city`."""
return self._utm_coordinates.y
"""The relative y-coordinate within the `.city` in meters.
On the implied x-y plane, the `.city`'s southwest corner is the origin.
Shortcut for `.location.y`.
"""
return self.location.y

View file

@ -1,6 +1,6 @@
"""Provide the ORM's `City` model."""
from typing import Any, Dict
from typing import Dict
import sqlalchemy as sa
from sqlalchemy import orm
@ -47,60 +47,52 @@ class City(meta.Base):
addresses = orm.relationship('Address', back_populates='city')
grids = orm.relationship('Grid', back_populates='city')
def __init__(self, *args: Any, **kwargs: Any) -> None:
"""Create a new city."""
# Call SQLAlchemy's default `.__init__()` method.
super().__init__(*args, **kwargs)
# Take the "lower left" of the viewport as the origin
# of a Cartesian coordinate system.
lower_left = self.viewport['southwest']
self._origin = utils.UTMCoordinate(
lower_left['latitude'], lower_left['longitude'],
)
# We do not implement a `.__init__()` method and leave that to SQLAlchemy.
# Instead, we use `hasattr()` to check for uninitialized attributes.
# grep:d334120e pylint:disable=attribute-defined-outside-init
def __repr__(self) -> str:
"""Non-literal text representation."""
return '<{cls}({name})>'.format(cls=self.__class__.__name__, name=self.name)
@property
def location(self) -> Dict[str, float]:
"""GPS location of the city's center.
def center(self) -> utils.Location:
"""Location of the city's center.
Example:
{"latitude": 48.856614, "longitude": 2.3522219}
Implementation detail: This property is cached as none of the
underlying attributes to calculate the value are to be changed.
"""
return {
'latitude': self._center_latitude,
'longitude': self._center_longitude,
}
if not hasattr(self, '_center'): # noqa:WPS421 note:d334120e
self._center = utils.Location(
self._center_latitude, self._center_longitude,
)
return self._center
@property
def viewport(self) -> Dict[str, Dict[str, float]]:
def viewport(self) -> Dict[str, utils.Location]:
"""Google Maps viewport of the city.
Example:
{
'northeast': {'latitude': 48.9021449, 'longitude': 2.4699208},
'southwest': {'latitude': 48.815573, 'longitude': 2.225193},
Implementation detail: This property is cached as none of the
underlying attributes to calculate the value are to be changed.
"""
if not hasattr(self, '_viewport'): # noqa:WPS421 note:d334120e
self._viewport = {
'northeast': utils.Location(
self._northeast_latitude, self._northeast_longitude,
),
'southwest': utils.Location(
self._southwest_latitude, self._southwest_longitude,
),
}
""" # noqa:RST203
return {
'northeast': {
'latitude': self._northeast_latitude,
'longitude': self._northeast_longitude,
},
'southwest': {
'latitude': self._southwest_latitude,
'longitude': self._southwest_longitude,
},
}
return self._viewport
@property
def as_origin(self) -> utils.UTMCoordinate:
"""The lower left corner of the `.viewport` in UTM coordinates.
def as_xy_origin(self) -> utils.Location:
"""The southwest corner of the `.viewport`.
This property serves as the `relative_to` argument to the
`UTMConstructor` when representing an `Address` in the x-y plane.
This property serves, for example, as the `other` argument to the
`Location.relate_to()` method when representing an `Address`
in the x-y plane.
"""
return self._origin
return self.viewport['southwest']

View file

@ -1,3 +1,3 @@
"""Utilities used by the ORM models."""
from urban_meal_delivery.db.utils.coordinates import UTMCoordinate # noqa:F401
from urban_meal_delivery.db.utils.locations import Location

View file

@ -1,4 +1,4 @@
"""A `UTMCoordinate` class to unify working with coordinates."""
"""A `Location` class to unify working with coordinates."""
from __future__ import annotations
@ -7,19 +7,27 @@ from typing import Optional
import utm
class UTMCoordinate:
"""A GPS location represented in UTM coordinates.
class Location:
"""A location represented in WGS84 and UTM coordinates.
For further info, we refer to this comprehensive article on the UTM system:
https://en.wikipedia.org/wiki/Universal_Transverse_Mercator_coordinate_system
WGS84:
- "conventional" system with latitude-longitude pairs
- assumes earth is a sphere and models the location in 3D
UTM:
- the Universal Transverse Mercator sytem
- projects WGS84 coordinates onto a 2D map
- can be used for visualizations and calculations directly
- distances are in meters
Further info how WGS84 and UTM are related:
https://en.wikipedia.org/wiki/Universal_Transverse_Mercator_coordinate_system
"""
# pylint:disable=too-many-instance-attributes
def __init__(
self, latitude: float, longitude: float, relative_to: UTMCoordinate = None,
) -> None:
"""Cast a WGS84-conforming `latitude`-`longitude` pair as UTM coordinates."""
def __init__(self, latitude: float, longitude: float) -> None:
"""Create a location from a WGS84-conforming `latitude`-`longitude` pair."""
# The SQLAlchemy columns come as `Decimal`s due to the `DOUBLE_PRECISION`.
self._latitude = float(latitude)
self._longitude = float(longitude)
@ -35,36 +43,40 @@ class UTMCoordinate:
self._normalized_easting: Optional[int] = None
self._normalized_northing: Optional[int] = None
if relative_to:
try:
self.relate_to(relative_to)
except TypeError:
raise TypeError(
'`relative_to` must be a `UTMCoordinate` object',
) from None
except ValueError:
raise ValueError(
'`relative_to` must be in the same UTM zone as the `latitude`-`longitude` pair', # noqa:E501
) from None
def __repr__(self) -> str:
"""A non-literal text representation.
"""A non-literal text representation in the UTM system.
Convention is {ZONE} {EASTING} {NORTHING}.
Example:
`<UTM: 17T 630084 4833438>'`
`<Location: 17T 630084 4833438>'`
"""
return f'<UTM: {self.zone} {self.easting} {self.northing}>' # noqa:WPS221
return f'<Location: {self.zone} {self.easting} {self.northing}>' # noqa:WPS221
@property
def latitude(self) -> float:
"""The latitude of the location in degrees (WGS84).
Between -90 and +90 degrees.
"""
return self._latitude
@property
def longitude(self) -> float:
"""The longitude of the location in degrees (WGS84).
Between -180 and +180 degrees.
"""
return self._longitude
@property
def easting(self) -> int:
"""The easting of the location in meters."""
"""The easting of the location in meters (UTM)."""
return self._easting
@property
def northing(self) -> int:
"""The northing of the location in meters."""
"""The northing of the location in meters (UTM)."""
return self._northing
@property
@ -73,8 +85,8 @@ class UTMCoordinate:
return f'{self._zone}{self._band}'
def __eq__(self, other: object) -> bool:
"""Check if two `UTMCoordinate` objects are the same location."""
if not isinstance(other, UTMCoordinate):
"""Check if two `Location` objects are the same location."""
if not isinstance(other, Location):
return NotImplemented
if self.zone != other.zone:
@ -104,24 +116,21 @@ class UTMCoordinate:
return self._normalized_northing
def relate_to(self, other: UTMCoordinate) -> None:
def relate_to(self, other: Location) -> None:
"""Make the origin in the lower-left corner relative to `other`.
The `.x` and `.y` properties are the `.easting` and `.northing` values
of `self` minus the ones from `other`. So, `.x` and `.y` make up a
Cartesian coordinate system where the `other` origin is `(0, 0)`.
This method is implicitly called by `.__init__()` if that is called
with a `relative_to` argument.
To prevent semantic errors in calculations based on the `.x` and `.y`
properties, the `other` origin may only be set once!
"""
if self._normalized_easting is not None:
raise RuntimeError('the `other` origin may only be set once')
if not isinstance(other, UTMCoordinate):
raise TypeError('`other` is not a `UTMCoordinate` object')
if not isinstance(other, Location):
raise TypeError('`other` is not a `Location` object')
if self.zone != other.zone:
raise ValueError('`other` must be in the same zone, including the band')