Source code for winterthur.views.streets

from itertools import groupby
from onegov.core.cache import lru_cache
from onegov.core.security import Public, Private
from onegov.winterthur import WinterthurApp, _
from onegov.winterthur.collections import AddressCollection
from onegov.winterthur.collections import AddressSubsetCollection
from onegov.winterthur.layout import AddressLayout
from onegov.winterthur.layout import AddressSubsetLayout
from urllib.parse import quote_plus


from typing import TYPE_CHECKING
if TYPE_CHECKING:
    from onegov.core.types import RenderData
    from onegov.winterthur.models import WinterthurAddress
    from onegov.winterthur.request import WinterthurRequest


@WinterthurApp.html(
    model=AddressCollection,
    permission=Public,
    template='streets.pt'
)


def view_streets(
    self: AddressCollection,
    request: 'WinterthurRequest'
) -> 'RenderData':

    request.include('street-search')

    by_letter = {
        letter: tuple(streets) for letter, streets in groupby(
            self.streets(), lambda s: s.letter
        )
    }

    @lru_cache(maxsize=1)
    def link_to_street(street: str) -> str:
        return request.class_link(AddressSubsetCollection, {'street': street})

    return {
        'layout': AddressLayout(self, request),
        'title': _('Streets Directory'),
        'streets': by_letter,
        'link_to_street': link_to_street,
        'last_updated': self.last_updated(),
        'update_state': self.update_state()
    }



@WinterthurApp.view(
    model=AddressCollection,
    permission=Private,
    request_method='POST',
    name='update'
)


def update_streets(
    self: AddressCollection,
    request: 'WinterthurRequest'
) -> None:

    request.assert_valid_csrf_token()

    self.update()

    request.success(_('The streets directory has been updated'))



@WinterthurApp.html(
    model=AddressSubsetCollection,
    permission=Public,
    template='street.pt'
)


def view_street(
    self: AddressSubsetCollection,
    request: 'WinterthurRequest'
) -> 'RenderData':

    def external_link_to_street(address: 'WinterthurAddress') -> str:
        q = quote_plus(str(address.street_id))
        return f'https://stadtplan.winterthur.ch/?locate=strasse&locations={q}'

    def external_link_to_address(address: 'WinterthurAddress') -> str:
        q = quote_plus(str(address.title))
        return f'https://stadtplan.winterthur.ch/?locate=adresse&locations={q}'

    return {
        'layout': AddressSubsetLayout(self, request),
        'title': self.street,
        'addresses': self.subset(),
        'parent': request.class_link(AddressCollection),
        'external_link_to_street': external_link_to_street,
        'external_link_to_address': external_link_to_address
    }