Daemon

Complexity

Total Complexity

17

Size/Duplication

Total Lines	72
Duplicated Lines	0 %

Importance

Changes	5
Bugs	0	Features	1

9 Methods

Rating	Name	Duplication	Size	Complexity
A	connect()	0	7	3
A	quit()	0	3	1
A	connect_external()	0	8	2
A	start()	0	3	1
A	__init__()	0	10	1
B	login()	0	23	5
A	connect_internal()	0	2	1
A	stop()	0	2	1
A	request()	0	4	2

#!/usr/bin/env python

import os
import time
import socket

from pprint import pprint

from heppy.EPP import REPP
from heppy.Error import Error
from heppy.Login import Login
from heppy.Client import Client
from heppy.Request import Request
from heppy.Response import Response

class Daemon:
    def __init__(self, config):
        self.config = config
        self.is_external = False
        self.client = None
        self.handler = SignalHandler({
            'SIGINT':  self.quit,
            'SIGTERM': self.quit,
            'SIGHUP':  self.hello,
            'SIGUSR1': self.hello,
            'SIGUSR2': self.hello,
        })

    def quit(self):
        global quit
        quit()

    def start(self,args = {}):
        self.connect()
        self.login(args)

    def connect(self):
        if self.client is not None:
            return
        if self.is_external:
            self.connect_external()
        else:
            self.connect_internal()

    def login(self, args = {}):
        greeting = self.client.get_greeting()
        greetobj = Response.parsexml(greeting)
        pprint(greetobj.data)
        try:
            request = Login.build(self.config, greeting, args)
            query = str(request)
            print Request.prettifyxml(query)
            reply = self.client.request(query)
            print Request.prettifyxml(reply)
        except Error as e:
            Error.die(2, 'failed perform login request')
        error = None
        try:
            response = Response.parsexml(reply)
            data = response.data
            pprint(data)
        except Error as e:
            error = e.message
            data = e.data
        if error is not None and data['resultCode']!='2002':
            Error.die(2, 'bad login response', data)
        print 'LOGIN OK'

    def connect_internal(self):
        self.client = REPP(self.config['epp'])

    def connect_external(self):
        try:
            self.client = Client(self.config['local']['address'])
            self.client.connect()
        except socket.error as e:
            os.system(self.config['zdir'] + '/eppyd ' + self.config['path'] + ' &')
            time.sleep(2)
            self.client = Client(self.config['local']['address'])

    def stop(self, args = {}):
        Error.die(3, 'stop not implemented', config)

    def request(self, query):
        with self.handler.block_signals():
            reply = self.client.request(query)
        return reply