X-Git-Url: https://git.toastfreeware.priv.at/philipp/winterrodeln/wrpylib.git/blobdiff_plain/f64a954a8596d88ee77707d2c42892aba161b385..d0bc0d2c5f2e975e49695d26b1afb24a648ead2f:/wrpylib/wrmwmarkup.py?ds=sidebyside diff --git a/wrpylib/wrmwmarkup.py b/wrpylib/wrmwmarkup.py index bfab0ab..0e299fa 100644 --- a/wrpylib/wrmwmarkup.py +++ b/wrpylib/wrmwmarkup.py @@ -2,19 +2,114 @@ # -*- coding: iso-8859-15 -*- # $Id$ # $HeadURL$ -"""This module contains winterrodeln specific functions that are prcocessing the MediaWiki markup. +"""This module contains winterrodeln specific functions that are processing the MediaWiki markup. """ import re +import xml.etree.ElementTree +import collections import formencode import wrpylib.wrvalidators import wrpylib.mwmarkup +WRMAP_POINT_TYPES = ['gasthaus', 'haltestelle', 'parkplatz', 'achtung', 'foto', 'verleih', 'punkt'] +WRMAP_LINE_TYPES = ['rodelbahn', 'gehweg', 'alternative', 'lift', 'anfahrt', 'linie'] -def _conv(fnct, value, fieldname): - """Internal function. - Like one of the to_xxx functions (e.g. to_bool), but adds the field name to the error message""" - try: return fnct(value) - except formencode.Invalid as e: raise formencode.Invalid(u"Conversion error in field '%s': %s" % (fieldname, unicode(e)), e.value, e.state) + +class ParseError(RuntimeError): + """Exception used by some of the functions""" + pass + + +class RodelbahnboxDictConverter(formencode.Validator): + """Converts a dict with Rodelbahnbox properties to a Sledrun class. Does no validation.""" + + def to_python(self, value, state=None): + """value is a dict of properties. If state is an object with the attribute sledrun, this sledrun class will be populated or updated.""" + props = value + if isinstance(state, object) and hasattr(state, 'sledrun'): + sledrun = state.sledrun + else: + class Sledrun(object): + pass + sledrun = Sledrun() + for k, v in props.iteritems(): + if k == u'Position': sledrun.position_latitude, sledrun.position_longitude = v + elif k == u'Position oben': sledrun.top_latitude, sledrun.top_longitude = v + elif k == u'Höhe oben': sledrun.top_elevation = v + elif k == u'Position unten': sledrun.bottom_latitude, sledrun.bottom_longitude = v + elif k == u'Höhe unten': sledrun.bottom_elevation = v + elif k == u'Länge': sledrun.length = v + elif k == u'Schwierigkeit': sledrun.difficulty = v + elif k == u'Lawinen': sledrun.avalanches = v + elif k == u'Betreiber': sledrun.operator = v + elif k == u'Öffentliche Anreise': sledrun.public_transport = v + elif k == u'Aufstieg möglich': sledrun.walkup_possible = v + elif k == u'Aufstieg getrennt': sledrun.walkup_separate, sledrun.walkup_separate_comment = v + elif k == u'Gehzeit': sledrun.walkup_time = v + elif k == u'Aufstiegshilfe': sledrun.lift, sledrun.lift_details = v + elif k == u'Beleuchtungsanlage': sledrun.night_light, sledrun.night_light_comment = v + elif k == u'Beleuchtungstage': sledrun.night_light_days, sledrun.night_light_days_comment = v + elif k == u'Rodelverleih': sledrun.sled_rental, sledrun.sled_rental_comment = v + elif k == u'Gütesiegel': sledrun.cachet = v + elif k == u'Webauskunft': sledrun.information_web = v + elif k == u'Telefonauskunft': sledrun.information_phone = v + elif k == u'Bild': sledrun.image = v + elif k == u'In Übersichtskarte': sledrun.show_in_overview = v + elif k == u'Forumid': sledrun.forum_id = v + return sledrun + + def from_python(self, value, state=None): + """Converts a sledrun class to a dict of Rodelbahnbox properties. value is a sledrun instance.""" + sledrun = value + r = collections.OrderedDict() + r[u'Position'] = (sledrun.position_latitude, sledrun.position_longitude) + r[u'Position oben'] = (sledrun.top_latitude, sledrun.top_longitude) + r[u'Höhe oben'] = sledrun.top_elevation + r[u'Position unten'] = (sledrun.bottom_latitude, sledrun.bottom_longitude) + r[u'Höhe unten'] = sledrun.bottom_elevation + r[u'Länge'] = sledrun.length + r[u'Schwierigkeit'] = sledrun.difficulty + r[u'Lawinen'] = sledrun.avalanches + r[u'Betreiber'] = sledrun.operator + r[u'Öffentliche Anreise'] = sledrun.public_transport + r[u'Aufstieg möglich'] = sledrun.walkup_possible + r[u'Aufstieg getrennt'] = (sledrun.walkup_separate, sledrun.walkup_separate_comment) + r[u'Gehzeit'] = sledrun.walkup_time + r[u'Aufstiegshilfe'] = (sledrun.lift, sledrun.lift_details) + r[u'Beleuchtungsanlage'] = (sledrun.night_light, sledrun.night_light_comment) + r[u'Beleuchtungstage'] = (sledrun.night_light_days, sledrun.night_light_days_comment) + r[u'Rodelverleih'] = (sledrun.sled_rental, sledrun.sled_rental_comment) + r[u'Gütesiegel'] = sledrun.cachet + r[u'Webauskunft'] = sledrun.information_web + r[u'Telefonauskunft'] = sledrun.information_phone + r[u'Bild'] = sledrun.image + r[u'In Übersichtskarte'] = sledrun.show_in_overview + r[u'Forumid'] = sledrun.forum_id + return r + + +class WinterrodelnTemplateDict(formencode.Validator): + """Private helper class for RodelbahnboxValidator or GasthausboxValidator""" + def __init__(self, template_title): + self.template_title = template_title + + def to_python(self, value, state): + title, anonym_params, named_params = value + if title != self.template_title: + raise formencode.Invalid(u'Template title has to be "{}".'.format(self.template_title), value, state) + if len(anonym_params) > 0: + raise formencode.Invalid(u'No anonymous parameters are allowed in "{}".'.format(self.template_title), value, state) + return named_params + + def from_python(self, value, state): + return self.template_title, [], value + + +class RodelbahnboxValidator(wrpylib.wrvalidators.RodelbahnboxDictValidator): + def __init__(self): + wrpylib.wrvalidators.RodelbahnboxDictValidator.__init__(self) + self.pre_validators=[wrpylib.mwmarkup.TemplateValidator(as_table=True, as_table_keylen=20), WinterrodelnTemplateDict(u'Rodelbahnbox')] + self.chained_validators = [RodelbahnboxDictConverter()] def rodelbahnbox_to_sledrun(wikitext, sledrun=None): @@ -22,167 +117,110 @@ def rodelbahnbox_to_sledrun(wikitext, sledrun=None): to a sledrun. sledrun may be an instance of WrSledrunCache or an "empty" class (object()) (default). Raises a formencode.Invalid exception if the format is not OK or the Rodelbahnbox is not found. :return: (start, end, sledrun) tuple of the Rodelbahnbox.""" - if sledrun is None: - class Sledrun(object): pass - sledrun = Sledrun() - - # match Rodelbahnbox + # find Rodelbahnbox start, end = wrpylib.mwmarkup.find_template(wikitext, u'Rodelbahnbox') if start is None: raise formencode.Invalid(u"Rodelbahnbox nicht gefunden", wikitext, None) - template_title, properties = wrpylib.mwmarkup.split_template(wikitext[start:end]) - - # process properties - for key, value in properties.iteritems(): - if key == u'Position': sledrun.position_latitude, sledrun.position_longitude = _conv(wrpylib.wrvalidators.GeoNone().to_python, value, key) # '47.583333 N 15.75 E' - elif key == u'Position oben': sledrun.top_latitude, sledrun.top_longitude = _conv(wrpylib.wrvalidators.GeoNone().to_python, value, key) # '47.583333 N 15.75 E' - elif key == u'Höhe oben': sledrun.top_elevation = _conv(wrpylib.wrvalidators.UnsignedNone().to_python, value, key) # '2000' - elif key == u'Position unten': sledrun.bottom_latitude, sledrun.bottom_longitude = _conv(wrpylib.wrvalidators.GeoNone().to_python, value, key) # '47.583333 N 15.75 E' - elif key == u'Höhe unten': sledrun.bottom_elevation = _conv(wrpylib.wrvalidators.UnsignedNone().to_python, value, key) # '1200' - elif key == u'Länge': sledrun.length = _conv(wrpylib.wrvalidators.UnsignedNone().to_python, value, key) # 3500 - elif key == u'Schwierigkeit': sledrun.difficulty = _conv(wrpylib.wrvalidators.GermanDifficulty().to_python, value, key) # 'mittel' elif key == u'Lawinen': sledrun.avalanches = _conv(wrpylib.wrvalidators.GermanAvalanches().to_python, value, key) # 'kaum' - elif key == u'Lawinen': sledrun.avalanches = _conv(wrpylib.wrvalidators.GermanAvalanches().to_python, value, key) # 'kaum' - elif key == u'Betreiber': sledrun.operator = _conv(wrpylib.wrvalidators.UnicodeNone().to_python, value, key) # 'Max Mustermann' - elif key == u'Öffentliche Anreise': sledrun.public_transport = _conv(wrpylib.wrvalidators.GermanPublicTransport().to_python, value, key) # 'Mittelmäßig' - elif key == u'Aufstieg möglich': sledrun.walkup_possible = _conv(wrpylib.wrvalidators.GermanBoolNone().to_python, value, key) # 'Ja' - elif key == u'Aufstieg getrennt': sledrun.walkup_separate, sledrun.walkup_separate_comment = _conv(wrpylib.wrvalidators.GermanTristateFloatComment().to_python, value, key) # 'Ja' - elif key == u'Gehzeit': sledrun.walkup_time = _conv(wrpylib.wrvalidators.UnsignedNone().to_python, value, key) # 90 - elif key == u'Aufstiegshilfe': sledrun.lift, sledrun.lift_details = _conv(wrpylib.wrvalidators.GermanLift().to_python, value, key) # 'Gondel (unterer Teil)' - elif key == u'Beleuchtungsanlage': sledrun.night_light, sledrun.night_light_comment = _conv(wrpylib.wrvalidators.GermanTristateFloatComment().to_python, value, key) - elif key == u'Beleuchtungstage': sledrun.night_light_days, sledrun.night_light_days_comment = _conv(wrpylib.wrvalidators.UnsignedCommentNone(7).to_python, value, key) # '3 (Montag, Mittwoch, Freitag)' - elif key == u'Rodelverleih': sledrun.sled_rental, sledrun.sled_rental_comment = _conv(wrpylib.wrvalidators.SledRental().to_python, value, key) # 'Talstation Serlesbahnan' - elif key == u'Gütesiegel': sledrun.cachet = _conv(wrpylib.wrvalidators.GermanCachet().to_python, value, key) # 'Tiroler Naturrodelbahn-Gütesiegel 2009 mittel' - elif key == u'Webauskunft': sledrun.information_web = _conv(wrpylib.wrvalidators.UrlNeinNone().to_python, value, key) # 'http://www.nösslachhütte.at/page9.php' - elif key == u'Telefonauskunft': sledrun.information_phone = _conv(wrpylib.wrvalidators.PhoneCommentListNeinLoopNone(comments_are_optional=False).to_python, value, key) # '+43-664-5487520 (Mitterer Alm)' - elif key == u'Bild': sledrun.image = _conv(wrpylib.wrvalidators.UnicodeNone().to_python, value, key) - elif key == u'In Übersichtskarte': sledrun.show_in_overview = _conv(wrpylib.wrvalidators.GermanBoolNone().to_python, value, key) - elif key == u'Forumid': sledrun.forum_id = _conv(wrpylib.wrvalidators.UnsignedNeinNone().to_python, value, key) - else: raise formencode.Invalid(u"Unbekannte Eigenschaft der Rodelbahnbox: '%s' (mit Wert '%s')" % (key, value), value, None) - return start, end, sledrun - - -def sledrun_to_rodelbahnbox(sledrun, version): + + # convert to sledrun + if sledrun is None: + state = None + else: + class State(object): + pass + state = State() + state.sledrun = sledrun + return start, end, RodelbahnboxValidator().to_python(wikitext[start:end], state) + + +def sledrun_to_rodelbahnbox(sledrun, version=None): """Converts a sledrun class to the {{Rodelbahnbox}} representation. The sledrun class has to have properties like position_latitude, ... See the table sledruncache for field (column) values. :param sledrun: an arbitrary class that contains the right properties :param version: a string specifying the version of the rodelbahnbox zu produce. - Version '1.3' and '1.4' are supported.""" - keys = [] - values = [] - keys.append(u'Position') - values.append(wrpylib.wrvalidators.GeoNone().from_python((sledrun.position_latitude, sledrun.position_longitude))) - keys.append(u'Position oben') - values.append(wrpylib.wrvalidators.GeoNone().from_python((sledrun.top_latitude, sledrun.top_longitude))) - keys.append(u'Höhe oben') - values.append(wrpylib.wrvalidators.UnsignedNone().from_python(sledrun.top_elevation)) - keys.append(u'Position unten') - values.append(wrpylib.wrvalidators.GeoNone().from_python((sledrun.bottom_latitude, sledrun.bottom_longitude))) - keys.append(u'Höhe unten') - values.append(wrpylib.wrvalidators.UnsignedNone().from_python(sledrun.bottom_elevation)) - keys.append(u'Länge') - values.append(wrpylib.wrvalidators.UnsignedNone().from_python(sledrun.length)) - keys.append(u'Schwierigkeit') - values.append(wrpylib.wrvalidators.GermanDifficulty().from_python(sledrun.difficulty)) - keys.append(u'Lawinen') - values.append(wrpylib.wrvalidators.GermanAvalanches().from_python(sledrun.avalanches)) - keys.append(u'Betreiber') - values.append(wrpylib.wrvalidators.UnicodeNone().from_python(sledrun.operator)) - keys.append(u'Öffentliche Anreise') - values.append(wrpylib.wrvalidators.GermanPublicTransport().from_python(sledrun.public_transport)) - if version == '1.4': - keys.append(u'Aufstieg möglich') - values.append(wrpylib.wrvalidators.GermanBoolNone().from_python(sledrun.walkup_possible)) - keys.append(u'Aufstieg getrennt') - values.append(wrpylib.wrvalidators.GermanTristateFloatComment().from_python((sledrun.walkup_separate, sledrun.walkup_separate_comment))) - keys.append(u'Gehzeit') - values.append(wrpylib.wrvalidators.UnsignedNone().from_python(sledrun.walkup_time)) - keys.append(u'Aufstiegshilfe') - values.append(wrpylib.wrvalidators.GermanLift().from_python((sledrun.lift, sledrun.lift_details))) - keys.append(u'Beleuchtungsanlage') - values.append(wrpylib.wrvalidators.GermanTristateFloatComment().from_python((sledrun.night_light, sledrun.night_light_comment))) - keys.append(u'Beleuchtungstage') - values.append(wrpylib.wrvalidators.UnsignedCommentNone(max=7).from_python((sledrun.night_light_days, sledrun.night_light_days_comment))) - keys.append(u'Rodelverleih') - values.append(wrpylib.wrvalidators.SledRental().from_python((sledrun.sled_rental, sledrun.sled_rental_comment))) - keys.append(u'Gütesiegel') - values.append(wrpylib.wrvalidators.GermanCachet().from_python(sledrun.cachet)) - keys.append(u'Webauskunft') - values.append(wrpylib.wrvalidators.UrlNeinNone().from_python(sledrun.information_web)) - keys.append(u'Telefonauskunft') - values.append(wrpylib.wrvalidators.PhoneCommentListNeinLoopNone(comments_are_optional=False).from_python(sledrun.information_phone)) - keys.append(u'Bild') - values.append(wrpylib.wrvalidators.UnicodeNone().from_python(sledrun.image)) - keys.append(u'In Übersichtskarte') - values.append(wrpylib.wrvalidators.GermanBoolNone().from_python(sledrun.show_in_overview)) - keys.append(u'Forumid') - values.append(wrpylib.wrvalidators.UnsignedNeinNone().from_python(sledrun.forum_id)) - return wrpylib.mwmarkup.create_template(u'Rodelbahnbox', [], keys, values, True, 20) + Version '1.4' is supported.""" + assert version in [None, '1.4'] + return RodelbahnboxValidator().from_python(sledrun) + + +class GasthausboxDictConverter(formencode.Validator): + """Converts a dict with Gasthausbox properties to a Inn class. Does no validation.""" + + def to_python(self, value, state=None): + """value is a dict of properties. If state is an object with the attribute inn, this inn class will be populated or updated.""" + props = value + if isinstance(state, object) and hasattr(state, 'inn'): + sledrun = state.sledrun + else: + class Inn(object): + pass + inn = Inn() + for k, v in props.iteritems(): + if k == u'Position': inn.position_latitude, inn.position_longitude = v + elif k == u'Höhe': inn.position_elevation = v + elif k == u'Betreiber': inn.operator = v + elif k == u'Sitzplätze': inn.seats = v + elif k == u'Übernachtung': inn.overnight, inn.overnight_comment = v + elif k == u'Rauchfrei': inn.nonsmoker_area, inn.smoker_area = v + elif k == u'Rodelverleih': inn.sled_rental, inn.sled_rental_comment = v + elif k == u'Handyempfang': inn.mobile_provider = v + elif k == u'Homepage': inn.homepage = v + elif k == u'E-Mail': inn.email_list = v + elif k == u'Telefon': inn.phone_list = v + elif k == u'Bild': inn.image = v + elif k == u'Rodelbahnen': inn.sledding_list = v + return inn + + def from_python(self, value, state=None): + """Converts an inn class to a dict of Gasthausbox properties. value is an Inn instance.""" + inn = value + r = collections.OrderedDict() + r[u'Position'] = (inn.position_latitude, inn.position_longitude) + r[u'Höhe'] = inn.position_elevation + r[u'Betreiber'] = inn.operator + r[u'Sitzplätze'] = inn.seats + r[u'Übernachtung'] = (inn.overnight, inn.overnight_comment) + r[u'Rauchfrei'] = (inn.nonsmoker_area, inn.smoker_area) + r[u'Rodelverleih'] = (inn.sled_rental, inn.sled_rental_comment) + r[u'Handyempfang'] = inn.mobile_provider + r[u'Homepage'] = inn.homepage + r[u'E-Mail'] = inn.email_list + r[u'Telefon'] = inn.phone_list + r[u'Bild'] = inn.image + r[u'Rodelbahnen'] = inn.sledding_list + return r + + +class GasthausboxValidator(wrpylib.wrvalidators.GasthausboxDictValidator): + def __init__(self): + wrpylib.wrvalidators.GasthausboxDictValidator.__init__(self) + self.pre_validators=[wrpylib.mwmarkup.TemplateValidator(as_table=True, as_table_keylen=17), WinterrodelnTemplateDict(u'Gasthausbox')] + self.chained_validators = [GasthausboxDictConverter()] def gasthausbox_to_inn(wikitext, inn=None): """Converts a inn wiki page containing a {{Gasthausbox}} to an inn. - raises a formencode.Invalid exception if an error occurs. + inn may be an instance of WrInnCache or an "empty" class (default). + raises a formencode.Invalid exception if the format is not OK or the Gasthausbox is not found. :return: (start, end, inn) tuple.""" - if inn is None: - class Inn(object): pass - inn = Inn() - - # Match Gasthausbox + # find Gasthausbox start, end = wrpylib.mwmarkup.find_template(wikitext, u'Gasthausbox') if start is None: raise formencode.Invalid(u"No 'Gasthausbox' found", wikitext, None) - template_title, properties = wrpylib.mwmarkup.split_template(wikitext[start:end]) - - # Process properties - for key, value in properties.iteritems(): - if key == u'Position': inn.position_latitude, inn.position_longitude = _conv(wrpylib.wrvalidators.GeoNone().to_python, value, key) # '47.583333 N 15.75 E' - elif key == u'Höhe': inn.position_elevation = _conv(wrpylib.wrvalidators.UnsignedNone().to_python, value, key) - elif key == u'Betreiber': inn.operator = _conv(wrpylib.wrvalidators.UnicodeNone().to_python, value, key) - elif key == u'Sitzplätze': inn.seats = _conv(wrpylib.wrvalidators.UnsignedNone().to_python, value, key) - elif key == u'Übernachtung': inn.overnight, inn.overnight_comment = _conv(wrpylib.wrvalidators.BoolUnicodeTupleValidator().to_python, value, key) - elif key == u'Rauchfrei': inn.nonsmoker_area, inn.smoker_area = _conv(wrpylib.wrvalidators.GermanTristateTuple().to_python, value, key) - elif key == u'Rodelverleih': inn.sled_rental, inn.sled_rental_comment = _conv(wrpylib.wrvalidators.BoolUnicodeTupleValidator().to_python, value, key) - elif key == u'Handyempfang': inn.mobile_provider = _conv(wrpylib.wrvalidators.ValueCommentListNeinLoopNone().to_python, value, key) - elif key == u'Homepage': inn.homepage = _conv(wrpylib.wrvalidators.UrlNeinNone().to_python, value, key) - elif key == u'E-Mail': inn.email_list = _conv(wrpylib.wrvalidators.EmailCommentListNeinLoopNone(allow_masked_email=True).to_python, value, key) - elif key == u'Telefon': inn.phone_list = _conv(wrpylib.wrvalidators.PhoneCommentListNeinLoopNone(comments_are_optional=True).to_python, value, key) - elif key == u'Bild': inn.image = _conv(wrpylib.wrvalidators.UnicodeNone().to_python, value, key) - elif key == u'Rodelbahnen': inn.sledding_list = _conv(wrpylib.wrvalidators.WikiPageListLoopNone().to_python, value, key) - else: raise formencode.Invalid(u"Unbekannte Eigenschaft der Gasthausbox: '%s' (mit Wert '%s')" % (key, value), value, None) - return start, end, inn + + # convert to inn + if inn is None: + state = None + else: + class State(object): + pass + state = State() + state.inn = inn + return start, end, GasthausboxValidator().to_python(wikitext[start:end], state) def inn_to_gasthausbox(inn): """Converts the inn class to the {{Gasthausbox}} representation.""" - keys = [] - values = [] - keys.append(u'Position') - values.append(wrpylib.wrvalidators.GeoNone().from_python((inn.position_latitude, inn.position_longitude))) - keys.append(u'Höhe') - values.append(wrpylib.wrvalidators.UnsignedNone().from_python(inn.position_elevation)) - keys.append(u'Betreiber') - values.append(wrpylib.wrvalidators.UnicodeNone().from_python(inn.operator)) - keys.append(u'Sitzplätze') - values.append(wrpylib.wrvalidators.UnsignedNone().from_python(inn.seats)) - keys.append(u'Übernachtung') - values.append(wrpylib.wrvalidators.BoolUnicodeTupleValidator().from_python((inn.overnight, inn.overnight_comment))) - keys.append(u'Rauchfrei') - values.append(wrpylib.wrvalidators.GermanTristateTuple().from_python((inn.nonsmoker_area, inn.smoker_area))) - keys.append(u'Rodelverleih') - values.append(wrpylib.wrvalidators.BoolUnicodeTupleValidator().from_python((inn.sled_rental, inn.sled_rental_comment))) - keys.append(u'Handyempfang') - values.append(wrpylib.wrvalidators.ValueCommentListNeinLoopNone().from_python(inn.mobile_provider)) - keys.append(u'Homepage') - values.append(wrpylib.wrvalidators.UrlNeinNone().from_python(inn.homepage)) - keys.append(u'E-Mail') - values.append(wrpylib.wrvalidators.EmailCommentListNeinLoopNone(allow_masked_email=True).from_python(inn.email_list)) - keys.append(u'Telefon') - values.append(wrpylib.wrvalidators.PhoneCommentListNeinLoopNone(comments_are_optional=True).from_python(inn.phone_list)) - keys.append(u'Bild') - values.append(wrpylib.wrvalidators.UnicodeNone().from_python(inn.image)) - keys.append(u'Rodelbahnen') - values.append(wrpylib.wrvalidators.WikiPageListLoopNone().from_python(inn.sledding_list)) - result = [u'{{Gasthausbox'] - return wrpylib.mwmarkup.create_template(u'Gasthausbox', [], keys, values, True) + return GasthausboxValidator().from_python(inn) def find_template_latlon_ele(wikitext, template_title): @@ -314,3 +352,205 @@ def find_all_templates(wikitext, find_func): result = (start, end) + result[2:] return results + +def googlemap_to_wrmap(attributes, coords, paths): + """Converts the output of parse_googlemap to the GeoJSON format wrmap uses. + :returns: (GeoJSON as nested Python datatypes) + """ + json_features = [] + + # point + for point in coords: + lon, lat, symbol, title = point + properties = {'type': 'punkt' if symbol is None else symbol.lower()} + if title is not None: properties['name'] = title + json_features.append({ + 'type': 'Feature', + 'geometry': {'type': 'Point', 'coordinates': [lon, lat]}, + 'properties': properties}) + + # path + for path in paths: + style, entries = path + style = style.lower() + PATH_TYPES = {u'6#ff014e9a': u'rodelbahn', u'6#ffe98401': u'gehweg', u'6#ff7f7fff': u'alternative', u'3#ff000000': u'lift', u'3#ffe1e100': u'anfahrt'} + if PATH_TYPES.has_key(style): + properties = {'type': PATH_TYPES[style]} + else: + properties = {'type': 'line'} + properties['dicke'] = style[0] + properties['farbe'] = style[4:] + json_features.append({ + 'type': 'Feature', + 'geometry': { + 'type': 'LineString', + 'coordinates': [[lon, lat] for lon, lat, symbol, title in entries]}, + 'properties': properties}) + + geojson = { + 'type': 'FeatureCollection', + 'features': json_features, + 'properties': attributes} + return geojson + + +def parse_wrmap_coordinates(coords): + '''gets a string coordinates and returns an array of lon/lat coordinate pairs, e.g. + 47.12 N 11.87 E + 47.13 N 11.70 E + -> + [[11.87, 47.12], [11.70, 47.13]]''' + result = [] + pos = 0 + for match in re.finditer(r'\s*(\d+\.?\d*)\s*N?\s+(\d+\.?\d*)\s*E?\s*', coords): + if match.start() != pos: + break + result.append([float(match.groups()[1]), float(match.groups()[0])]) + pos = match.end() + else: + if pos == len(coords): + return result + raise RuntimeError('Wrong coordinate format: {}'.format(coords)) + + +def parse_wrmap(wikitext): + """Parses the (unicode) u'content' of the Winterrodeln wrmap extension. + If wikitext does not contain the tag or if the tag contains + invalid formatted lines, a ParseError is raised. + Use wrpylib.mwmarkup.find_tag(wikitext, 'wrmap') to find the wrmap tag within an arbitrary + wikitext before using this function. + + :param wikitext: wikitext containing only the template. Example: + + wikitext = u''' + + 47.240689 11.190454 + 47.245789 11.238971 + 47.245711 11.238283 + + 47.238587 11.203360 + 47.244951 11.230868 + 47.245470 11.237853 + + + ''' + :returns: GeoJSON as nested Python datatype + """ + # parse XML + try: + wrmap_xml = xml.etree.ElementTree.fromstring(wikitext.encode('utf-8')) + except xml.etree.ElementTree.ParseError as e: + row, column = e.position + raise ParseError("XML parse error on row {}, column {}: {}".format(row, column, e)) + if wrmap_xml.tag not in ['wrmap', 'wrgmap']: + raise ParseError('No valid tag name') + + # convert XML to geojson (http://www.geojson.org/geojson-spec.html) + json_features = [] + for feature in wrmap_xml: + # determine feature type + is_point = feature.tag in WRMAP_POINT_TYPES + is_line = feature.tag in WRMAP_LINE_TYPES + if (not is_point and not is_line): + raise ParseError('Unknown element <{}>.'.format(feature.tag)) + + # point + if is_point: + properties = {'type': feature.tag} + allowed_properties = set(['name', 'wiki']) + wrong_properties = set(feature.attrib.keys()) - allowed_properties + if len(wrong_properties) > 0: + raise ParseError("The attribute '{}' is not allowed at <{}>.".format(list(wrong_properties)[0], feature.tag)) + properties.update(feature.attrib) + coordinates = parse_wrmap_coordinates(feature.text) + if len(coordinates) != 1: + raise ParseError('The element <{}> has to have exactly one coordinate pair.'.format(feature.tag)) + json_features.append({ + 'type': 'Feature', + 'geometry': {'type': 'Point', 'coordinates': coordinates[0]}, + 'properties': properties}) + + # line + if is_line: + properties = {'type': feature.tag} + allowed_properties = set(['farbe', 'dicke']) + wrong_properties = set(feature.attrib.keys()) - allowed_properties + if len(wrong_properties) > 0: + raise ParseError("The attribute '{}' is not allowed at <{}>.".format(list(wrong_properties)[0], feature.tag)) + if feature.attrib.has_key('farbe'): + if not re.match('#[0-9a-fA-F]{6}$', feature.attrib['farbe']): + raise ParseError('The attribute "farbe" has to have a format like "#a0bb43".') + properties['strokeColor'] = feature.attrib['farbe'] # e.g. #a200b7 + if feature.attrib.has_key('dicke'): + try: + properties['strokeWidth'] = int(feature.attrib['dicke']) # e.g. 6 + except ValueError: + raise ParseError('The attribute "dicke" has to be an integer.') + json_features.append({ + 'type': 'Feature', + 'geometry': {'type': 'LineString', 'coordinates': parse_wrmap_coordinates(feature.text)}, + 'properties': properties}) + + # attributes + properties = {} + for k, v in wrmap_xml.attrib.iteritems(): + if k in ['lat', 'lon']: + try: + properties[k] = float(v) + except ValueError: + raise ParseError('Attribute "{}" has to be a float value.'.format(k)) + elif k in ['zoom', 'width', 'height']: + try: + properties[k] = int(v) + except ValueError: + raise ParseError('Attribute "{}" has to be an integer value.'.format(k)) + else: + raise ParseError('Unknown attribute "{}".'.format(k)) + + geojson = { + 'type': 'FeatureCollection', + 'features': json_features, + 'properties': properties} + + return geojson + + +def create_wrmap_coordinates(coords): + result = [] + for coord in coords: + result.append('{:.6f} N {:.6f} E'.format(coord[1], coord[0])) + return '\n'.join(result) + + +def create_wrmap(geojson): + """Creates a wikitext from geojson (as python types).""" + wrmap_xml = xml.etree.ElementTree.Element('wrmap') + wrmap_xml.text = '\n\n' + for k, v in geojson['properties'].iteritems(): + if k in ['lon', 'lat']: + wrmap_xml.attrib[k] = '{:.6f}'.format(v) + else: + wrmap_xml.attrib[k] = str(v) + + assert geojson['type'] == 'FeatureCollection' + json_features = geojson['features'] + last_json_feature = None + for json_feature in json_features: + feature_xml = xml.etree.ElementTree.SubElement(wrmap_xml, json_feature['properties']['type']) + geo = json_feature['geometry'] + if geo['type'] == 'Point': + feature_xml.text = create_wrmap_coordinates([geo['coordinates']]) + if last_json_feature is not None: + last_json_feature.tail = '\n' + else: + if last_json_feature is not None: + last_json_feature.tail = '\n\n' + feature_xml.text = '\n' + create_wrmap_coordinates(geo['coordinates']) + '\n' + last_json_feature = feature_xml + feature_xml.attrib = json_feature['properties'] + del feature_xml.attrib['type'] + + if last_json_feature is not None: + last_json_feature.tail = '\n\n' + return xml.etree.ElementTree.tostring(wrmap_xml, encoding='utf-8').decode('utf-8') +