You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
3.1 KiB
Python
103 lines
3.1 KiB
Python
2 years ago
|
import os, time
|
||
|
|
|
||
|
|
'''
|
||
|
|
Plugin handler for the Reftek export plugin.
|
||
|
|
'''
|
||
|
|
class SeedlinkPluginHandler:
|
||
|
|
# Create defaults
|
||
|
|
def __init__(self):
|
||
|
|
self.instances = {}
|
||
|
|
self.unitMap = {}
|
||
|
|
self.idMap = {}
|
||
|
|
|
||
|
|
def push(self, seedlink):
|
||
|
|
# Check and set defaults
|
||
|
|
address = "127.0.0.1"
|
||
|
|
try: address = seedlink.param('sources.reftek.address')
|
||
|
|
except: seedlink.setParam('sources.reftek.address', address)
|
||
|
|
|
||
|
|
port = 2543
|
||
|
|
try: port = int(seedlink.param('sources.reftek.port'))
|
||
|
|
except: seedlink.setParam('sources.reftek.port', port)
|
||
|
|
|
||
|
|
proc = "reftek"
|
||
|
|
try: proc = seedlink.param('sources.reftek.proc')
|
||
|
|
except: seedlink.setParam('sources.reftek.proc', proc)
|
||
|
|
|
||
|
|
key = (address + ":" + str(port), proc)
|
||
|
|
|
||
|
|
try:
|
||
|
|
reftekId = self.instances[key]
|
||
|
|
|
||
|
|
except KeyError:
|
||
|
|
reftekId = len(self.instances)
|
||
|
|
self.instances[key] = reftekId
|
||
|
|
self.unitMap[reftekId] = []
|
||
|
|
|
||
|
|
try:
|
||
|
|
unit = seedlink.param('sources.reftek.unit')
|
||
|
|
map = os.path.join(seedlink.config_dir, "reftek2sl%d.map" % reftekId)
|
||
|
|
seedlink.setParam('sources.reftek.mapFlag', map)
|
||
|
|
|
||
|
|
mapping = (unit, seedlink._get('seedlink.station.id'))
|
||
|
|
if not mapping in self.unitMap[reftekId]:
|
||
|
|
self.unitMap[reftekId].append(mapping)
|
||
|
|
|
||
|
|
try:
|
||
|
|
if not mapping[1] in self.idMap[mapping[0]]:
|
||
|
|
self.idMap[mapping[0]].append(mapping[1])
|
||
|
|
except KeyError:
|
||
|
|
self.idMap[mapping[0]] = [mapping[1]]
|
||
|
|
|
||
|
|
except KeyError:
|
||
|
|
try:
|
||
|
|
map = seedlink.param('sources.reftek.map')
|
||
|
|
if not os.path.isabs(map):
|
||
|
|
map = os.path.join(seedlink.config_dir, map)
|
||
|
|
except: map = os.path.join(seedlink.config_dir, 'reftek2sl.map')
|
||
|
|
seedlink.setParam('sources.reftek.mapFlag', map)
|
||
|
|
|
||
|
|
timeout = 60
|
||
|
|
try: timeout = int(seedlink.param('sources.reftek.timeout'))
|
||
|
|
except: seedlink.setParam('sources.reftek.timeout', timeout)
|
||
|
|
|
||
|
|
default_tq = 40
|
||
|
|
try: default_tq = int(seedlink.param('sources.reftek.default_tq'))
|
||
|
|
except: seedlink.setParam('sources.reftek.default_tq', default_tq)
|
||
|
|
|
||
|
|
unlock_tq = 10
|
||
|
|
try: unlock_tq = int(seedlink.param('sources.reftek.unlock_tq'))
|
||
|
|
except: seedlink.setParam('sources.reftek.unlock_tq', unlock_tq)
|
||
|
|
|
||
|
|
log_soh = "true"
|
||
|
|
try:
|
||
|
|
if seedlink.param('sources.reftek.log_soh').lower() in ("yes", "true", "1"):
|
||
|
|
log_soh = "true"
|
||
|
|
else:
|
||
|
|
log_soh = "false"
|
||
|
|
except: seedlink.setParam('sources.reftek.log_soh', log_soh)
|
||
|
|
|
||
|
|
# Key is address (one instance per address)
|
||
|
|
return (key, map)
|
||
|
|
|
||
|
|
|
||
|
|
# Flush generates the map file(s)
|
||
|
|
def flush(self, seedlink):
|
||
|
|
for x in self.idMap.keys():
|
||
|
|
if len(self.idMap[x]) > 1:
|
||
|
|
raise Exception("Error: Reftek plugin has multiple mappings for unit %s" % x)
|
||
|
|
|
||
|
|
for x in self.unitMap.keys():
|
||
|
|
mappings = self.unitMap[x]
|
||
|
|
if len(mappings) == 0: continue
|
||
|
|
|
||
|
|
reftek2slmap = os.path.join(seedlink.config_dir, "reftek2sl%d.map" % x)
|
||
|
|
|
||
|
|
fd = open(reftek2slmap, "w")
|
||
|
|
|
||
|
|
fd.write("# Generated at %s - Do not edit!\n" % time.strftime("%Y-%m-%d %H:%M:%S UTC", time.gmtime()))
|
||
|
|
for c in mappings:
|
||
|
|
fd.write("%s %s\n" % (c[0], c[1]))
|
||
|
|
|
||
|
|
fd.close()
|