Source code for httpwriter

#!/usr/bin/env python
"""
.. module:: httpwriter.py
   :platform: Unix, Windows
   :synopsis: Ulyxes - an open source project to drive total stations and
       publish observation results.
       GPL v2.0 license
       Copyright (C) 2010- Zoltan Siki <siki.zoltan@epito.bme.hu>

.. moduleauthor:: Zoltan Siki <siki.zoltan@epito.bme.hu>
"""

import logging
try:
    from urllib.request import urlopen, Request # for python 3
    from urllib.parse import urlencode
except ImportError:
    from urllib import urlencode
    from urllib2 import urlopen, Request        # for python 2
from angle import Angle
from writer import Writer

class HttpWriter(Writer):
    """ Class to write observations to a web server (HTTP GET/POST)

            :param name: name of writer (str)
            :param angle: angle unit to use (str)
            :param dist: distance and coordinate format (str)
            :param dt: date/time format (str), default ansi
            :param filt: list of allowed keys (list)
            :param url: url to server side script (str)
            :param mode: GET/POST
    """

    def __init__(self, name=None, angle='GON', dist='.3f',
                 dt='%Y-%m-%d %H:%M:%S', filt=None,
                 url='http://localhost/monitoring/get.php', mode='GET'):
        """ Constructor
        """
        super().__init__(name, angle, dist, dt, filt)
        self.url = url
        self.mode = mode

    def WriteData(self, data):
        """ Write observation data to server

            :param data: dictionary with observation data
            :returns: server answer or None (if error)
        """
        par = {}
        if data is None or self.DropData(data):
            logging.warning(" empty or inappropiate data not written")
            return -1
        # add datetime and/or id
        data = self.ExtendData(data)
        for key, val in data.items():
            if self.filt is None or key in self.filt:
                par[key] = self.StrVal(val)
        if self.mode == 'GET':
            try:
                res = urlopen(self.url + '?' + urlencode(par)).read()
            except Exception:
                res = -1
        else:
            try:
                d = urlencode(par).encode('ascii')
                req = Request(self.url, d)
                res = urlopen(req).read()
            except Exception:
                res = -1
        return int(res)

if __name__ == "__main__":
    #myfile = HttpWriter(mode='GET', url="http://www.agt.bme.hu/php/get_tester.php")
    myfile = HttpWriter(mode='POST', url="http://www.agt.bme.hu/php/get_tester.php")
    v = Angle(100.2345, 'GON')
    data = {'id': '1', 'hz': Angle(0.12345), 'v': Angle(100.2365, 'GON'), 'distance': 123.6581}
    print(myfile.WriteData(data))