87 pyro.initServer()

   169 

    88 daemon=pyro.Daemon(host=ip, port=port)

   170 class Server():

    89 uri = daemon.connect(PLCObject(WorkingDir, daemon, args),"PLCObject")

   171     def __init__(self, name, ip, port, workdir, args):

    90 

   172         self.continueloop = True

    91 print "The daemon runs on port :",daemon.port

   173         self.daemon = None

    92 print "The object's uri is :",uri

   174         self.name = name

    93 print "The working directory :",WorkingDir

   175         self.ip = ip

    94 

   176         self.port = port

    95 # Configure and publish service

   177         self.workdir = workdir

    96 # Not publish service if localhost in address params

   178         self.args = args

    97 if ip != "localhost" and ip != "127.0.0.1":    

   179         self.plcobj = None

    98     print "Publish service on local network"

   180         self.servicepublisher = ServicePublisher.ServicePublisher()

    99     service = ServicePublisher.ServicePublisher(name, ip, port)

   181 

   100 

   182     def Loop(self):

   101 sys.stdout.flush()

   183         while self.continueloop:

   102 

   184             self.Start()

   103 daemon.requestLoop()

   185         

   186     def Restart(self):

   187         self.Stop()

   188 

   189     def Quit(self):

   190         self.continueloop = False

   191         self.Stop()

   192 

   193     def Start(self):

   194         pyro.initServer()

   195         self.daemon=pyro.Daemon(host=self.ip, port=self.port)

   196         self.plcobj = PLCObject(self.workdir, self.daemon, self.args)

   197         uri = self.daemon.connect(self.plcobj,"PLCObject")

   198     

   199         print "The daemon runs on port :",self.port

   200         print "The object's uri is :",uri

   201         print "The working directory :",self.workdir

   202         

   203         # Configure and publish service

   204         # Not publish service if localhost in address params

   205         if self.ip != "localhost" and self.ip != "127.0.0.1":    

   206             print "Publish service on local network"            

   207             self.servicepublisher.RegisterService(self.name, self.ip, self.port)

   208         

   209         sys.stdout.flush()

   210         

   211         self.daemon.requestLoop()

   212     

   213     def Stop(self):

   214         self.servicepublisher.UnRegisterService()

   215         self.daemon.shutdown(True)

   216 

   217 class ParamsEntryDialog(wx.TextEntryDialog):

   218     if wx.VERSION < (2, 6, 0):

   219         def Bind(self, event, function, id = None):

   220             if id is not None:

   221                 event(self, id, function)

   222             else:

   223                 event(self, function)

   224     

   225 

   226     def __init__(self, parent, message, caption = "Please enter text", defaultValue = "", 

   227                        style = wx.OK|wx.CANCEL|wx.CENTRE, pos = wx.DefaultPosition):

   228         wx.TextEntryDialog.__init__(self, parent, message, caption, defaultValue, style, pos)

   229         

   230         self.Tests = []

   231         if wx.VERSION >= (2, 8, 0):

   232             self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetAffirmativeId())

   233         elif wx.VERSION >= (2, 6, 0):

   234             self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetAffirmativeButton().GetId())

   235         else:

   236             self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetChildren()[0].GetSizer().GetChildren()[0].GetWindow().GetId())

   237     

   238     def OnOK(self, event):

   239         value = self.GetValue()

   240         texts = {"value" : value}

   241         for function, message in self.Tests:

   242             if not function(value):

   243                 message = wx.MessageDialog(self, message%texts, "Error", wx.OK|wx.ICON_ERROR)

   244                 message.ShowModal()

   245                 message.Destroy()

   246                 return

   247         self.EndModal(wx.ID_OK)

   248     

   249     def GetValue(self):

   250         return self.GetSizer().GetItem(1).GetWindow().GetValue()

   251     

   252     def SetTests(self, tests):

   253         self.Tests = tests

   254         

   255 class DemoTaskBarIcon(wx.TaskBarIcon):

   256     TBMENU_CHANGE_NAME = wx.NewId()

   257     TBMENU_CHANGE_PORT = wx.NewId()

   258     TBMENU_CHANGE_INTERFACE = wx.NewId()

   259     TBMENU_CHANGE_WD = wx.NewId()

   260     TBMENU_QUIT = wx.NewId()

   261     

   262     def __init__(self, pyroserver):

   263         wx.TaskBarIcon.__init__(self)

   264         # Set the image

   265         icon = self.MakeIcon(BeremizIcon.GetImage())

   266         self.SetIcon(icon, "Beremiz Service")        

   267         # bind some events

   268         #self.Bind(wx.EVT_TASKBAR_CLICK, self.OnClick)

   269         #self.Bind(wx.EVT_TASKBAR_LEFT_DCLICK, self.OnTaskBarActivate)

   270         self.Bind(wx.EVT_MENU, self.OnTaskBarChangeName, id=self.TBMENU_CHANGE_NAME)

   271         self.Bind(wx.EVT_MENU, self.OnTaskBarChangeInterface, id=self.TBMENU_CHANGE_INTERFACE)

   272         self.Bind(wx.EVT_MENU, self.OnTaskBarChangePort, id=self.TBMENU_CHANGE_PORT)

   273         self.Bind(wx.EVT_MENU, self.OnTaskBarChangeWorkingDir, id=self.TBMENU_CHANGE_WD)

   274         self.Bind(wx.EVT_MENU, self.OnTaskBarQuit, id=self.TBMENU_QUIT)

   275     

   276     def CreatePopupMenu(self):

   277         """

   278         This method is called by the base class when it needs to popup

   279         the menu for the default EVT_RIGHT_DOWN event.  Just create

   280         the menu how you want it and return it from this function,

   281         the base class takes care of the rest.

   282         """

   283         menu = wx.Menu()

   284         menu.Append(self.TBMENU_CHANGE_NAME, "Change Name")

   285         menu.Append(self.TBMENU_CHANGE_INTERFACE, "Change IP of interface to bind")

   286         menu.Append(self.TBMENU_CHANGE_PORT, "Change Port Number")

   287         menu.AppendSeparator()

   288         menu.Append(self.TBMENU_CHANGE_WD, "Change working directory")

   289         menu.Append(self.TBMENU_QUIT, "Quit")

   290         return menu

   291 

   292     def MakeIcon(self, img):

   293         """

   294         The various platforms have different requirements for the

   295         icon size...

   296         """

   297         if "wxMSW" in wx.PlatformInfo:

   298             img = img.Scale(16, 16)

   299         elif "wxGTK" in wx.PlatformInfo:

   300             img = img.Scale(22, 22)

   301         # wxMac can be any size upto 128x128, so leave the source img alone....

   302         icon = wx.IconFromBitmap(img.ConvertToBitmap() )

   303         return icon

   304     

   305     def OnTaskBarChangeInterface(self,evt):

   306         dlg = ParamsEntryDialog(None, "Enter the ip of the interface to bind", defaultValue=pyroserver.ip)

   307         dlg.SetTests([(re.compile('\d{1,3}(?:\.\d{1,3}){3}$').match, "Ip is not valid!"),

   308                        ( lambda ip :len([x for x in ip.split(".") if 0 <= int(x) <= 255]) == 4, "Ip is not valid!")

   309                        ])

   310         if dlg.ShowModal() == wx.ID_OK:

   311             pyroserver.ip = dlg.GetValue()

   312             pyroserver.Stop()

   313             

   314     def OnTaskBarChangePort(self,evt):

   315         dlg = ParamsEntryDialog(None, "Enter a port number ", defaultValue=str(pyroserver.port))

   316         dlg.SetTests([(UnicodeType.isdigit, "Port number must be an integer!"), (lambda port : 0 <= int(port) <= 65535 , "Port number must be 0 <= port <= 65535!")])

   317         if dlg.ShowModal() == wx.ID_OK:

   318             pyroserver.port = int(dlg.GetValue())

   319             pyroserver.Stop()

   320             

   321     

   322     def OnTaskBarChangeWorkingDir(self,evt):

   323         dlg = wx.DirDialog(None, "Choose a working directory ", pyroserver.workdir, wx.DD_NEW_DIR_BUTTON)

   324         if dlg.ShowModal() == wx.ID_OK:

   325             pyroserver.workdir = dlg.GetPath()

   326             pyroserver.Stop()

   327             

   328     def OnTaskBarChangeName(self,evt):

   329         dlg = ParamsEntryDialog(None, "Enter a name ", defaultValue=pyroserver.name)

   330         dlg.SetTests([(lambda name : len(name) is not 0 , "Name must not be null!")])

   331         if dlg.ShowModal() == wx.ID_OK:

   332             pyroserver.name = dlg.GetValue()

   333             pyroserver.Restart()

   334 

   335     def OnTaskBarQuit(self,evt):

   336         pyroserver.Quit()

   337         self.RemoveIcon()

   338         

   339 pyroserver = Server(name, ip, port, WorkingDir, args)

   340 

   341 if havewx:

   342     app=wx.App(redirect=False)

   343     taskbar_instance = DemoTaskBarIcon(pyroserver)

   344     pyro_thread=Thread(target=pyroserver.Loop)

   345     pyro_thread.start()

   346     app.MainLoop()

   347 else:

   348     pyroserver.Loop()