source: pvrrd/epics2rrd.py @ 236

#!/usr/bin/env python

'''
log EPICS data into RRD files
'''

########### SVN repository information ###################
# $Date: 2011-01-19 01:27:35 +0000 (Wed, 19 Jan 2011) $
# $Author: jemian $
# $Revision: 236 $
# $HeadURL$
# $Id: epics2rrd.py 236 2011-01-19 01:27:35Z jemian $
########### SVN repository information ###################

#-------------------------------------------------------------

import os
import time
import sys
import rrd_pv       # coordination between EPICS and RRD
import xmlSupport   # read pvlist.xml and properties.xml files

#-------------------------------------------------------------

svnrev = "$Revision: 236 $".strip("$").split()[1]

message = time.ctime()
message += " %s (r%s)" % ( sys.argv[0], svnrev )
message += " PID=" + repr(os.getpid())
message += " starting on HOST=" + os.environ['HOST']
message += " by user=" + os.environ['USER']
print "# ", message
sys.stdout.flush()

db = {}
cfg = xmlSupport.readConfigurationXML('/home/joule/WEB33/www/rrd')
properties = cfg['properties']

pvList = cfg['pvList']

for pvEntry in pvList:
    name = pvEntry['pv']
    if not name in db.keys():
        basedir = properties['rrd_logs_base_dir']
        rrdpath = properties['subdirs']['rrd_file_dir']
        rrdfile = pvEntry['fileRoot'] + '.rrd'
        rrd_database = os.path.join(basedir, rrdpath, rrdfile)

        fileRoot = pvEntry['fileRoot']
        pvObject = rrd_pv.rrd_pv(name, rrd_database)
        db[name] = pvObject

# throttle the printing of messages in the main event loop
nexttime = time.time()
update_interval_seconds = 10*60  # update every 10 minutes

while 1:
    pvcount = 0
    keys = db.keys()
    keys.sort()
    notconnected_msgs = []
    for name in keys:
        rrdPv = db[name]
        if rrdPv.pv.connected:
            pvcount += 1
            if rrdPv.pv.value == None:
                rrdPv.pv.get()
            rrdPv.update()
            #print rrdPv
        else:
            notconnected_msgs.append("Trying to connect: %s" % name)
            rrd_database = rrdPv.rrd
            #rrdPv.pv.ch.connect()  # does not seem to connect if initial try failed
            # replace existing object
            db[name] = rrd_pv.rrd_pv(name, rrd_database)
    if (time.time() >= nexttime):
        # throttle the frequency of this message:
        if len(notconnected_msgs):
            for msg in notconnected_msgs:
                print "# ", msg
        print "# ", time.ctime()," PID=" + repr(os.getpid()),
        print os.environ['USER'] + '@' + os.environ['HOST'],
        print "   %d of %d PVs connected" % (pvcount, len(db))
        nexttime = time.time() + update_interval_seconds  # set next update time
    sys.stdout.flush()
    time.sleep(5)