Source Code for Module ZenModel.IpServiceLoader

 1  ########################################################################### 
 2  # 
 3  # This program is part of Zenoss Core, an open source monitoring platform. 
 4  # Copyright (C) 2007, Zenoss Inc. 
 5  # 
 6  # This program is free software; you can redistribute it and/or modify it 
 7  # under the terms of the GNU General Public License version 2 as published by 
 8  # the Free Software Foundation. 
 9  # 
10  # For complete information please visit: http://www.zenoss.com/oss/ 
11  # 
12  ########################################################################### 
13   
14  __doc__ ='''IpServiceLoader 
15   
16  A script to load the IANA well known port numbers. 
17   
18  $Id: IpServiceLoader.py,v 1.5 2004/04/15 23:47:50 edahl Exp $''' 
19   
20  __version__ = "$Revision: 1.5 $"[11:-2] 
21   
22  import os 
23  import re 
24   
25  import Globals #initalize imports correctly magic! 
26   
27  from Products.ZenUtils.BasicLoader import BasicLoader 
28   
29  from Products.ZenModel.IpService import getIpServiceKey 
30  from Products.ZenModel.IpServiceClass import IpServiceClass 
31   
32 -class IpServiceLoader(BasicLoader): 
33   
34      lineparse = re.compile(r"^(\S+)\s+(\d+)/(\S+)(.*)") 
35           
36       
37 -    def __init__(self, noopts=0, app=None): 
38          self.filename = os.path.join(os.path.dirname(__file__),  
39                              "port-numbers.txt") 
40          BasicLoader.__init__(self, noopts, app, ignoreComments=False) 
41          services = self.dmd.getDmdRoot('Services') 
42          self.privserv = services.createOrganizer("/IpService/Privileged") 
43          self.regserv = services.createOrganizer("/IpService/Registered") 
44          self.lastservice = None 
45   
46   
47 -    def loaderBody(self, line): 
48          if line.startswith("#"): return 
49          m = self.lineparse.search(line) 
50          if not m: return 
51          keyword, port, proto, descr = m.groups() 
52          descr = descr.strip() 
53          port = int(port) 
54          svc = self.privserv.find(keyword) 
55          portkey = getIpServiceKey(proto, port) 
56          if not svc: 
57              serviceKeys = (keyword, portkey) 
58              svc = IpServiceClass(keyword, serviceKeys=serviceKeys, 
59                                 description=descr, port=port) 
60              if port < 1024: 
61                  self.privserv.serviceclasses._setObject(svc.id, svc) 
62              else: 
63                  self.regserv.serviceclasses._setObject(svc.id, svc) 
64              self.log.info("Added IpServiceClass %s" % keyword) 
65          else: 
66              svc.addServiceKey(portkey) 
67   
68   
69  if __name__ == "__main__": 
70      loader = IpServiceLoader() 
71      loader.loadDatabase() 
72      print "Database Load is finished!" 
73