From c9c7d389c45815e1b7abd7bea7dc5093a3630ea2 Mon Sep 17 00:00:00 2001 From: Gabriel Pérez-Cerezo Date: Tue, 10 Oct 2017 11:45:50 +0200 Subject: Moved stuff to a module folder, preparing for setuptools --- backends/studentenwerk.py | 72 ----------------------------------------------- 1 file changed, 72 deletions(-) delete mode 100644 backends/studentenwerk.py (limited to 'backends/studentenwerk.py') diff --git a/backends/studentenwerk.py b/backends/studentenwerk.py deleted file mode 100644 index 48b15d3..0000000 --- a/backends/studentenwerk.py +++ /dev/null @@ -1,72 +0,0 @@ -# Copyright (C) 2017 Gabriel Perez-Cerezo -# -*- coding: utf-8 -*- - -import urllib.request, urllib.error, urllib.parse -from lxml import etree -from lxml.cssselect import CSSSelector -import html5lib -from base import * -from yapsy.IPlugin import IPlugin -import multiprocessing - - -from yapsy import NormalizePluginNameForModuleName as normalize - -mensenliste = {"TU Hardenbergstraße" : "mensa-tu-hardenbergstra%C3%9Fe", "TU Marchstraße": "cafeteria-tu-marchstra%C3%9Fe", "TU Skyline": "cafeteria-tu-skyline", "TU Architektur": "cafeteria-tu-architektur", "TU Ackerstraße": "cafeteria-tu-ackerstra%C3%9Fe"} - - - -def pr_f(j) : - i,k = j - food = get_food_items(k, ignore_nudelauswahl=True) - return (i,"*"*20+i+"*"*20+"\n"+formt(food)) - -class Studentenwerk(IPlugin) : - def register_restaurants (self) : - mensenliste = {"TU Hardenbergstraße" : "mensa-tu-hardenbergstra%C3%9Fe", "TU Marchstraße": "cafeteria-tu-marchstra%C3%9Fe", "TU Skyline": "cafeteria-tu-skyline", "TU Architektur": "cafeteria-tu-architektur", "TU Ackerstraße": "cafeteria-tu-ackerstra%C3%9Fe"} - for h,n in mensenliste.items() : - r = Restaurant(normalize(h), h, self, "dummy", [n]) - register_restaurant(r) - def get_food_items(self, mensa="mensa-tu-hardenbergstra%C3%9Fe", ignore_nudelauswahl=False) : - user_agent = 'Mozilla/5.0 (Windows NT 6.1; Win64; x64)' - headers = {'User-Agent': user_agent} - - req = urllib.request.Request('https://www.stw.berlin/mensen/%s.html' % mensa, headers=headers) - response = urllib.request.urlopen(req) - the_page = response.read() - document = html5lib.parse(the_page, treebuilder="lxml") - groupsel = CSSSelector('.splGroupWrapper') - groups = [e for e in groupsel(document)] - fl = [] - for i in groups : - try: - name = CSSSelector('.splGroup')(i)[0].text - except: - raise NoMenuError from None - sel = CSSSelector('.splMeal') - meals = [e for e in sel(i)] - for m in meals : - namesel = CSSSelector('.bold') - nm = namesel(m)[0].text - if ignore_nudelauswahl and "Nudelauswahl" in nm : - continue - pricesel = CSSSelector('.col-md-3') - veg = 0 - if len(pricesel(m)[0]) >= 2 : - if "15" in pricesel(m)[0][1].attrib["src"] : - veg = 2 - elif "1.png" in pricesel(m)[0][1].attrib["src"] : - veg = 1 - price = pricesel(m)[-1].text.strip() - fl.append(Food(nm, price, name, veg)) - return fl - - - -# format: -# if __name__ == "__main__": -# pool = multiprocessing.Pool(4) -# k = pool.map(pr_f, list(mensenliste.items())) -# k.sort(key=lambda x: x[0]) -# for i in k : -# print(i[1]) -- cgit v1.2.3