sr0wx/meteoalarm_sq9atk.py

103 wiersze
3.7 KiB
Python
Executable File

#!/usr/bin/python -tt
# -*- coding: utf-8 -*-
import re
import logging
import socket
import requests
from pprint import pprint
from sr0wx_module import SR0WXModule
class MeteoalarmSq9atk(SR0WXModule):
"""Klasa pobierająca informacje z meteoalarm.pl"""
def __init__(self, language, service_url, region_id):
self.__service_url = service_url
self.__language = language
self.__region_id = region_id
self.__logger = logging.getLogger(__name__)
self.__levels = { '1' : 'niski', '2' : 'sredni', '3' : 'wysoki' }
self.__regions = {
'14':'mazowieckiego', '08':'lubuskiego', '32':'zachodniopomorskiego',
'22':'pomorskiego', '02':'dolnoslaskiego', '16':'opolskiego',
'24':'slaskiego', '12':'malopolskiego', '18':'podkarpackiego',
'26':'swietokrzyskiego', '10':'lodzkiego', '14':'wielkopolskiego',
'04':'kujawsko-pomorskiego', '28':'warminsko-mazurskiego', '06':'lubelskiego',
'20':'podlaskiego',
}
self.__warnings = {
'o01': 'burze', 'o02': 'deszcz_i_grad', 'o03': 'mgla',
'o06': 'ulewny_deszcz', 'o08': 'zamiec_sniezna', 'o04': 'marznaca_mgla',
'o11': 'snieg_lub_oblodzenie', 'o07': 'marznacy_deszcz', 'o08': 'sniezyca',
'o09': 'przymrozki', 'o10': 'roztopy', 'o05': 'niskie_temperatury',
'o13': 'silny_wiatr', 'o12': 'wysokie_temperatury', 'o14': 'zawieje_sniezne'
}
def getHtmlFromUrl(self, url):
try:
self.__logger.info("::: Odpytuję adres: " + url)
resp = requests.get(url, timeout=8)
if resp.status_code != 200:
self.__logger.error("::: Data response code error - %s \n" % resp.status_code)
return ''
return resp.content
except requests.exceptions.RequestException as e:
self.__logger.error("::: Data download error - %s \n" % e)
return ''
def findDataInHtml(self, html):
self.__logger.info("::: Przetwarzam dane...\n")
patternTable = re.compile(r'<table[^>]*class="[^>]*meteo_table[^>]*"[^>]*>.*?</table>', re.DOTALL)
matchTable = patternTable.search(html)
if matchTable:
tableHtml = matchTable.group(0)
levelMatch = re.search(r'<b>(\d+)</b>', tableHtml, re.DOTALL)
level = levelMatch.group(1) if levelMatch else None
imgMatch = re.search(r'<div class="zagrozenia-ikony">(.*?)</div>', tableHtml, re.DOTALL)
images = re.findall(r'<img.*?src="images/(.*?).png"', imgMatch.group(1)) if imgMatch else []
# deduplicate values
images = list(set(images))
# Wypisujemy nazwy plików zamiast atrybutów alt
return [level, images]
else:
return []
def get_data(self):
self.__logger.info("::: Pobieram dane o zagrożeniach...")
html = self.getHtmlFromUrl(self.__service_url + str(self.__region_id))
data = self.findDataInHtml(html)
message = ''
try:
level = data[0]
warnings = data[1]
message += " ".join([
' _ zagrozenia_meteorologiczne_dla_wojewodztwa ',
self.__regions[self.__region_id],
' _ ',
' '.join(self.__warnings[key] for key in warnings),
' _ poziom_zagrozenia ',
' '.join([self.__levels[level]]),
' _ '
])
except:
None
return {
"message": message,
"source": "meteoalarm_pl",
}