aboutsummaryrefslogtreecommitdiff
path: root/mensa/frontends/plain-text.py
blob: 57e1648494974fb7fb151fdd4c2d481667c66984 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
from mensa import base
from yapsy.IPlugin import IPlugin
import textwrap
class TextRenderer(IPlugin) :
    def render_line(self, item) :
        vegkeys = [ "", "Vegetarian", "Vegan" ]
        if self.options.get("only_veggie",0) > item.veggie :
            return ""
        desc = ""
        if item.desc : 
            desc = "      "+item.desc+"\n"
        nameparts = textwrap.wrap(item.name, 60)
        l = [ "  " + nameparts[0].ljust(60) + "    " + item.price.ljust(19) + vegkeys[item.veggie] ] + [ "  "+ i.ljust(50) for i in nameparts[1:]]
        # "\t" + item.name.ljust(80) + "\t"+ item.price.ljust(20) + vegkeys[item.veggie]+"\n"+desc
        return "\n".join(l) + "\n"
    
    def render (self, foods, **options) :
        self.options = options
        ## Expects list of tuples with (Restaurant, Foodlist)
        r = ""
        for restaurant, food in foods :
            cat = []
            if not food :
                continue
            r = r+"*"*20+restaurant.human_name+"*"*20+"\n"
            if "pos" in options and restaurant.pos and options.get("dist") :
                ## display distance to restaurant
                r=r+"Distance: %.2f km\n" % base.dist(options["pos"], restaurant.pos)

            r= r + "".join([category+"\n" + "".join([self.render_line(i) for i in items]) for category,items  in food.items()])

        if options.get("template") :
            f = open(options["template"], "r")
            template = f.read()
            f.close()
            print(template.replace("$$TEXT$$", r))
        else :
            print(r)

    def register_renderer(self) :
        base.register_renderer(base.Renderer("plain-text", "Plain Text Renderer", self))