1

#!/usr/bin/env python

     2

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

     3

     4

#This file is part of Beremiz, a Integrated Development Environment for

     5

#programming IEC 61131-3 automates supporting plcopen standard and CanFestival. 

     6

#

     7

#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD

     8

#

     9

#See COPYING file for copyrights details.

    10

#

    11

#This library is free software; you can redistribute it and/or

    12

#modify it under the terms of the GNU General Public

    13

#License as published by the Free Software Foundation; either

    14

#version 2.1 of the License, or (at your option) any later version.

    15

#

    16

#This library is distributed in the hope that it will be useful,

    17

#but WITHOUT ANY WARRANTY; without even the implied warranty of

    18

#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

    19

#General Public License for more details.

    20

#

    21

#You should have received a copy of the GNU General Public

    22

#License along with this library; if not, write to the Free Software

    23

#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

    24

    25

import os, sys, getopt

    26

    27

try:

    28

    import wx, re

    29

    from wx.lib.embeddedimage import PyEmbeddedImage

    30

    from threading import Thread

    31

    from types import *

    32

    havewx = True

    33

except:

    34

    havewx = False

    35

   127

   128

def usage():

   129

    print """

   130

Usage of Beremiz PLC execution service :\n

   131

%s {[-n name] [-i ip] [-p port]|-h|--help} working_dir

   132

           -n        - zeroconf service name

   133

           -i        - ip of interface to bind to (x.x.x.x)

   134

           -p        - port number

   135

           -h        - print this help text and quit

   136

   137

           working_dir - directory where are stored PLC files

   138

"""%sys.argv[0]

   139

   140

try:

   141

    opts, args = getopt.getopt(sys.argv[1:], "i:p:n:h")

   142

except getopt.GetoptError, err:

   143

    # print help information and exit:

   144

    print str(err) # will print something like "option -a not recognized"

   145

    usage()

   146

    sys.exit(2)

   147

   148

# default values

   149

ip = ""

   150

port = 3000

   151

name = os.environ[{

   152

     "linux2":"USER",

   153

     "win32":"USERNAME",

   154

     }.get(sys.platform, "USER")]

   155

   156

for o, a in opts:

   157

    if o == "-h":

   158

        usage()

   159

        sys.exit()

   160

    elif o == "-i":

   161

        if len(a.split(".")) == 4 or a == "localhost":

   162

            ip = a

   163

    elif o == "-p":

   164

        # port: port that the service runs on

   165

        port = int(a)

   166

    elif o == "-n":

   167

        name = a

   168

    else:

   169

        usage()

   170

        sys.exit()

   171

   172

if len(args) > 1:

   173

    usage()

   174

    sys.exit()

   175

elif len(args) == 1:

   176

    WorkingDir = args[0]

   177

elif len(args) == 0:

   178

    WorkingDir = os.getcwd()

   179

    args=[WorkingDir]

   180

   181

from runtime import PLCObject, ServicePublisher

   182

import Pyro.core as pyro

   183

   184

if not os.path.isdir(WorkingDir):

   185

    os.mkdir(WorkingDir)

   186

   187

   188

   189

class Server():

   190

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

   191

        self.continueloop = True

   192

        self.daemon = None

   193

        self.name = name

   194

        self.ip = ip

   195

        self.port = port

   196

        self.workdir = workdir

   197

        self.args = args

   198

        self.plcobj = None

   199

        self.servicepublisher = None

   204

        while self.continueloop:

   205

            self.Start()

   206

   207

    def Restart(self):

   208

        self.Stop()

   209

   210

    def Quit(self):

   211

        self.continueloop = False

   212

        self.Stop()

   213

   214

    def Start(self):

   215

        pyro.initServer()

   216

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

   218

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

   219

   220

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

   221

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

   222

        print "The working directory :",self.workdir

   223

   224

        # Configure and publish service

   225

        # Not publish service if localhost in address params

   226

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

   227

            print "Publish service on local network"

   228

            self.servicepublisher = ServicePublisher.ServicePublisher()            

   229

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

   230

   231

        sys.stdout.flush()

   232

   233

        self.daemon.requestLoop()

   234

   235

    def Stop(self):

   236

        if self.servicepublisher is not None:

   237

            self.servicepublisher.UnRegisterService()

   238

            del self.servicepublisher

   239

        self.daemon.shutdown(True)

   240

   241

class ParamsEntryDialog(wx.TextEntryDialog):

   242

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

   243

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

   244

            if id is not None:

   245

                event(self, id, function)

   246

            else:

   247

                event(self, function)

   248

   249

   250

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

   251

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

   252

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

   253

   254

        self.Tests = []

   255

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

   256

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

   257

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

   258

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

   259

        else:

   260

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

   261

   262

    def OnOK(self, event):

   263

        value = self.GetValue()

   264

        texts = {"value" : value}

   265

        for function, message in self.Tests:

   266

            if not function(value):

   267

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

   268

                message.ShowModal()

   269

                message.Destroy()

   270

                return

   271

        self.EndModal(wx.ID_OK)

   272

   273

    def GetValue(self):

   274

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

   275

   276

    def SetTests(self, tests):

   277

        self.Tests = tests

   278

   279

class DemoTaskBarIcon(wx.TaskBarIcon):

   280

    TBMENU_CHANGE_NAME = wx.NewId()

   281

    TBMENU_CHANGE_PORT = wx.NewId()

   282

    TBMENU_CHANGE_INTERFACE = wx.NewId()

   283

    TBMENU_CHANGE_WD = wx.NewId()

   284

    TBMENU_QUIT = wx.NewId()

   285

   286

    def __init__(self, pyroserver):

   287

        wx.TaskBarIcon.__init__(self)

   288

        # Set the image

   291

        # bind some events

   292

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

   293

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

   294

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

   295

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

   296

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

   297

   298

    def CreatePopupMenu(self):

   299

        """

   300

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

   301

        the menu for the default EVT_RIGHT_DOWN event.  Just create

   302

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

   303

        the base class takes care of the rest.

   304

        """

   305

        menu = wx.Menu()

   306

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

   307

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

   308

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

   309

        menu.AppendSeparator()

   310

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

   311

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

   312

        return menu

   313

   314

    def MakeIcon(self, img):

   315

        """

   316

        The various platforms have different requirements for the

   317

        icon size...

   318

        """

   319

        if "wxMSW" in wx.PlatformInfo:

   320

            img = img.Scale(16, 16)

   321

        elif "wxGTK" in wx.PlatformInfo:

   322

            img = img.Scale(22, 22)

   323

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

   324

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

   325

        return icon

   326

   327

    def OnTaskBarChangeInterface(self,evt):

   328

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

   329

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

   330

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

   331

                       ])

   332

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

   333

            pyroserver.ip = dlg.GetValue()

   334

            pyroserver.Stop()

   335

   336

    def OnTaskBarChangePort(self,evt):

   337

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

   338

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

   339

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

   340

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

   341

            pyroserver.Stop()

   342

   343

   344

    def OnTaskBarChangeWorkingDir(self,evt):

   345

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

   346

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

   347

            pyroserver.workdir = dlg.GetPath()

   348

            pyroserver.Stop()

   349

   350

    def OnTaskBarChangeName(self,evt):

   351

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

   352

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

   353

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

   354

            pyroserver.name = dlg.GetValue()

   355

            pyroserver.Restart()

   356

   357

    def OnTaskBarQuit(self,evt):

   358

        pyroserver.Quit()

   359

        self.RemoveIcon()

   360

        wx.GetApp().ExitMainLoop()

   361

   371

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

   372

   373

if havewx:

   374

    app=wx.App(redirect=False)

   375

    taskbar_instance = DemoTaskBarIcon(pyroserver)

   379

    pyro_thread.start()

   380

    app.MainLoop()

   381

else:

   382

    pyroserver.Loop()