1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     6

#

     8

#

    10

#

    15

#

    20

#

    24

    25

import socket, threading

    26

from util import Zeroconf

    27

    28

service_type = '_PYRO._tcp.local.'

    29

    30

class ServicePublisher():

    31

    def __init__(self):

    32

        # type: fully qualified service type name

    33

        self.serviceproperties = {'description':'Beremiz remote PLC'}

    34

    35

        self.name = None

    36

        self.ip_32b = None

    37

        self.port = None

    38

        self.server = None

    39

        self.service_name = None

    40

        self.retrytimer = None

    41

    42

    def RegisterService(self, name, ip, port):

    43

        try:

    44

            self._RegisterService(name, ip, port)

    45

        except Exception,e:

    46

            self.retrytimer = threading.Timer(2,self.RegisterService,[name, ip, port])

    47

            self.retrytimer.start() 

    48

    49

    def _RegisterService(self, name, ip, port):

    50

        # name: fully qualified service name

    51

        self.service_name = 'Beremiz_%s.%s'%(name,service_type)

    52

        self.name = name

    53

        self.port = port

    54

    55

        self.server = Zeroconf.Zeroconf(ip)

    56

        print "MDNS brodcasting on :"+ip

    57

    58

        if ip == "0.0.0.0":

    59

            ip = self.gethostaddr()

    60

        print "MDNS brodcasted service address :"+ip

    61

        self.ip_32b = socket.inet_aton(ip)

    62

    63

        self.server.registerService(

    64

             Zeroconf.ServiceInfo(service_type,

    65

                                  self.service_name,

    66

                                  self.ip_32b,

    67

                                  self.port,

    68

                                  properties = self.serviceproperties))

    69

        self.retrytimer=None

    70

    71

    def UnRegisterService(self):

    72

        if self.retrytimer is not None:

    73

            self.retrytimer.cancel()

    74

    75

        self.server.unregisterService(

    76

                                      Zeroconf.ServiceInfo(service_type, 

    77

                                                           self.service_name, 

    78

                                                           self.ip_32b, 

    79

                                                           self.port, 

    80

                                                           properties = self.serviceproperties))

    81

        self.server.close()

    82

        self.server = None

    83

    84

    def gethostaddr(self, dst = '224.0.1.41'):

    85

        s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

    86

        try:

    87

            s.connect((dst, 7))

    88

            (host, port) = s.getsockname()

    89

            s.close()

    90

            if host != '0.0.0.0':

    91

                return host

    92

        except Exception,e:

    93

            pass

    94

        return socket.gethostbyname(socket.gethostname())