Unifi wireless voucher generator

voucherator.py

#!/usr/bin/python -tt

# Copyright 2017 Garrett Holmstrom
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; either version 3 of the License or (at your
# option) any later version accepted by the Santa Barbara Hackerspace (or
# its successor approved by the Santa Barbara Hackerspace), which shall
# act as a proxy as defined in Section 14 of version 3 of the license.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see http://www.gnu.org/licenses/.

import argparse
import random
import time
import traceback

import pymongo

try:
    import BaseHTTPServer as httpserver
    import urlparse
except ImportError:
    import http.server as httpserver
    import urllib.parse as urlparse


CODE_DURATION_MINUTES = 1440
UNIFI_ADMIN_NAME = 'guestbutton'
UNIFI_SITE = 'FIXME'
UNIFI_NOTE = 'Wireless code button'


__version__ = '0.1.0'


class VoucherHTTPRequestHandler(httpserver.BaseHTTPRequestHandler):
    """
    Voucher generator HTTP request handler
    """

    protocol_version = 'HTTP/1.0'
    server_version = 'voucherator/{}'.format(
        '.'.join(__version__.split('.')[:2]))

    def do_POST(self):
        """
        Handle POST requests, the method for creating vouchers.
        """
        try:
            if self.path == '/voucher/new':
                voucher = gen_voucher()
                self.send_voucher(voucher)
            else:
                self.send_error(404)
        except:
            traceback.print_exc()
            self.send_error(500)

    def send_voucher(self, voucher):
        code = '{}-{}'.format(voucher['code'][:5], voucher['code'][5:])
        self.send_response(200)
        self.send_header('Connection', 'close')
        self.send_header('Content-Length', len(code))
        self.send_header('Content-Type', 'text/plain')
        self.end_headers()
        self.wfile.write(code)


def gen_voucher():
    start = time.time()
    end = start + CODE_DURATION_MINUTES * 60
    voucher = {
        'site_id': UNIFI_SITE,
        'create_time': 0,
        'code': '{0:0<10}'.format(random.randint(0, 9999999999)),
        'for_hotspot': False,
        'admin_name': UNIFI_ADMIN_NAME,
        'quota': 1,
        'duration': CODE_DURATION_MINUTES,
        'end_time': int(end),
        'used': 0,
        'qos_overwrite': False,
        'note': UNIFI_NOTE,
    }
    client = pymongo.MongoClient('127.0.0.1', 27117)
    db = client.ace
    db.voucher.insert(voucher)
    client.close()
    return voucher


def main():
    parser = argparse.ArgumentParser(
        description='Unifi wireless voucher generator')
    parser.add_argument('-b', '--bind-addr', default='0.0.0.0')
    parser.add_argument('-p', '--port', type=int, default=8080)
    args = parser.parse_args()
    httpd = httpserver.HTTPServer((args.bind_addr, args.port),
                                  VoucherHTTPRequestHandler)
    httpd.serve_forever()


if __name__ == '__main__':
    main()