fetch_reactor_stats.py

#!/usr/bin/env python
# coding: utf-8
#
# Copyright (c) 2015, Torbjörn Lönnemark <[email protected]>
#
# Permission to use, copy, modify, and/or distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
#
from __future__ import print_function, unicode_literals, division

import bs4
import pprint
import requests

class App:
    def fetch_oskarshamn(self):
        r = requests.get('http://okg.se/')
        soup = bs4.BeautifulSoup(r.content)

        reactors = ['o1', 'o2', 'o3']



        result = {}

        for reactor in reactors:
            rd = soup.find(id=reactor)
            re = rd.find(class_='value')
            prod = int(list(re.children)[0].strip())
            max_prod = int(''.join(rd.find('span').get_text().strip().split(' ')[:-1]))
            result[reactor.upper()] = {'production': prod, 'maxProduction': max_prod}

        return result

    def fetch_vattenfall(self, facility):
        r = requests.get('http://www2.vattenfall.se/miniapps/om_vattenfall/var_verksamhet/var_produktion/karnkraftsproduktion/index2.asp?verkIn={}&sizeIn=Small'.format(facility))
        soup = bs4.BeautifulSoup(r.content)

        percentages = soup.find_all('td', valign='bottom')[1:]
        productions = soup.find_all('td', valign=False)[2:-1]

        result = {}

        for perc, prod in zip(percentages, productions):
            reactor = prod.find('b').get_text().strip()
            utilization = float(perc.find('img').attrs['height'])/100
            prod_elem = prod.find('b').next_sibling
            while isinstance(prod_elem, bs4.NavigableString):
                prod_elem = prod_elem.next_sibling
            production = int(prod_elem.get_text().strip())
            max_production = None
            if production != 0:
                max_production = production/utilization

            result[reactor] = {'production': production, 'maxProduction': max_production}

        return result

    def fetch_all(self):
        res = {}
        res['ringhals'] = self.fetch_vattenfall('Ringhals')
        res['forsmark'] = self.fetch_vattenfall('Forsmark')
        res['oskarshamn'] = self.fetch_oskarshamn()
        return res

    def run(self):
        pprint.pprint(self.fetch_all())


if __name__ == '__main__':
    App().run()