Cuisine_Guess.py

#-*- coding: utf-8 -*-

###########################################################################
##                                                                       ##
## Copyrights Frédéric Rodrigo 2020                                      ##
##                                                                       ##
## This program is free software: you can redistribute it and/or modify  ##
## it under the terms of the GNU General Public License as published by  ##
## the Free Software Foundation, either version 3 of the License, or     ##
## (at your option) any later version.                                   ##
##                                                                       ##
## This program is distributed in the hope that it will be useful,       ##
## but WITHOUT ANY WARRANTY; without even the implied warranty of        ##
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         ##
## GNU General Public License for more details.                          ##
##                                                                       ##
## You should have received a copy of the GNU General Public License     ##
## along with this program.  If not, see <http://www.gnu.org/licenses/>. ##
##                                                                       ##
###########################################################################

from modules.OsmoseTranslation import T_
from plugins.Plugin import Plugin

from .modules import Cuisine_Guess_lib


class Cuisine_Guess(Plugin):

    only_for = ["FR"]

    def init(self, logger):
        Plugin.init(self, logger)
        detail = '''Using statistics based on amenity name, amenity tag value and takeaway tag, guess a possible value for `cuisine` tag.'''
        self.errors[1] = self.def_class(item = 3270, level = 3, tags = ['fix:survey'],
            title = T_('Possible mistake or lack of precision of `cuisine` value'),
            detail = T_(detail))
        self.errors[2] = self.def_class(item = 3270, level = 3, tags = ['fix:survey'],
            title = T_('Suggestion of `cuisine` value'),
            detail = T_(detail))

        self.taster = Cuisine_Guess_lib.Cuisine('dictionaries/Lang_fr/cuisine.csv')

    def node(self, data, tags):
        if 'name' not in tags or tags.get('amenity') not in ('restaurant', 'fast_food'):
            return

        cuisine_guess = self.taster.guess(tags['name'], tags['amenity'], tags.get('takeaway'))
        if cuisine_guess:
            tasty_cuisines = None
            if 'cuisine' in tags:
                max_score = max(map(lambda c: c[1], cuisine_guess.items()))
                if max_score < 0.9:
                    tasty_cuisines = True
                else:
                    cuisines = set(map(lambda s: s.strip(), tags['cuisine'].split(';')))
                    tasty_cuisines = cuisines.intersection(set(map(lambda c: c[0], cuisine_guess.items())))

            if not tasty_cuisines:
                return {'class': 1 if 'cuisine' in tags else 2,
                    'text': T_('Guess with probability: {0}', ', '.join(map(lambda cs: '{0} ({1}%)'.format(cs[0], round(cs[1] * 100)), cuisine_guess.items()))),
                    'fix': [{'~': {'cuisine': cuisine[0]}} for cuisine in cuisine_guess.items()]
                }

    def way(self, data, tags, nds):
        return self.node(data, tags)

    def relation(self, data, tags, members):
        return self.node(data, tags)


###########################################################################
from plugins.Plugin import TestPluginCommon

class Test(TestPluginCommon):
    def test(self):
        a = Cuisine_Guess(None)
        a.init(None)
        assert a.node(None, {"amenity": "restaurant", "name": "Fujiyama"})
        assert not a.node(None, {"amenity": "restaurant", "name": "lkgverjverkj"})