Skip to content
Snippets Groups Projects
Commit e4d28a3a authored by Auke Klazema's avatar Auke Klazema
Browse files

SW-598: Add port, username and password to load_expected_post.py

parent c572780c
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
MAC/Deployment/data/Coordinates/load_expected_pos.py
+ 104
− 62
View file @ e4d28a3a
#!/usr/bin/env python
#coding: iso-8859-15
import re,sys,pgdb,pg
from database import *
# coding: iso-8859-15
import re
import sys
import pgdb
import pg
from optparse import OptionParser
import getpass
# get info from database.py
dbName=getDBname()
dbHost=getDBhost()
#
# getCoordLines
#
def getCoordLines(filename):
def get_coord_lines(filename):
"""
Returns a list containing all lines with coordinates
"""
pattern=re.compile(r"^[HLC]{1}[0-9A-Z ]+,.*", re.IGNORECASE | re.MULTILINE)
#print pattern.findall(open(filename).read())
return [ line for line in pattern.findall(open(filename).read())]
pattern = re.compile(r"^[HLC]{1}[0-9A-Z ]+,.*", re.IGNORECASE | re.MULTILINE)
# print pattern.findall(open(filename).read())
return [line for line in pattern.findall(open(filename).read())]
#
......@@ -25,43 +25,82 @@ def getCoordLines(filename):
#
if __name__ == '__main__':
parser = OptionParser("Usage: %prog [options] datafile")
parser.add_option("-D", "--database",
dest="dbName",
type="string",
default="StationCoordinates",
help="Name of StationCoordinates database to use")
parser.add_option("-H", "--host",
dest="dbHost",
type="string",
default="sasdb.control.lofar",
help="Hostname of StationCoordinates database")
parser.add_option("-P", "--port",
dest="dbPort",
type="int",
default="5432",
help="Port of StationCoordinates database")
parser.add_option("-U", "--user",
dest="dbUser",
type="string",
default="postgres",
help="Username of StationCoordinates database")
# parse arguments
(options, args) = parser.parse_args()
if len(args) != 1:
parser.print_help()
sys.exit(0)
dbName = options.dbName
dbHost = options.dbHost
dbPort = options.dbPort
dbUser = options.dbUser
filename = args[0]
dbPassword = getpass.getpass()
# check syntax of invocation
# Expected syntax: load_measurement stationname objecttypes datafile
#
if (len(sys.argv) != 2):
print "Syntax: %s datafile" % sys.argv[0]
sys.exit(1)
filename = str(sys.argv[1])
stationname = filename[ filename.find('/')+1 : filename.find('/')+1 + 5].upper()
stationname = filename[filename.find('/') + 1: filename.find('/') + 1 + 5].upper()
objecttype = 'LBA, HBA'
refSys = 'ETRS89'
refSys = 'ETRS89'
refFrame = 'ETRF89'
method = 'derived'
date = '2010-01-01'
pers1 = 'Brentjens'
pers2 = 'Donker'
pers3 = ''
derived = ''
absRef = ''
comment = 'expected coordinates, Brentjens'
method = 'derived'
date = '2010-01-01'
pers1 = 'Brentjens'
pers2 = 'Donker'
pers3 = ''
derived = ''
absRef = ''
comment = 'expected coordinates, Brentjens'
# check some data against the database
station = []
station.append(stationname)
db = pgdb.connect(user="postgres", host=dbHost, database=dbName)
host = "{}:{}".format(dbHost, dbPort)
db = pgdb.connect(user=dbUser, host=host, database=dbName, password=dbPassword)
cursor = db.cursor()
# check person2
cursor.execute("select name from personnel where name = '%s'" % pers2 )
cursor.execute("select name from personnel where name = '%s'" % pers2)
if cursor.rowcount != 1:
print "Person: '%s' is not in the personnel file, add it (Y/N)?" % pers2
if raw_input().upper() == "Y":
insertcmd = db.cursor();
insertcmd = db.cursor()
insertcmd.execute("insert into personnel values ('%s')" % pers2)
db.commit()
else:
sys.exit(1);
sys.exit(1)
# check stationname
cursor.execute("select name from station")
stations = cursor.fetchall()
......@@ -84,37 +123,39 @@ if __name__ == '__main__':
print 'absolute reference : ', absRef
print 'comment : ', comment
#if raw_input('Continue processing this file (Y/N)?').upper() != "Y":
# if raw_input('Continue processing this file (Y/N)?').upper() != "Y":
# sys.exit(1)
print 'processing ',
sys.stdout.flush()
# calling stored procedures only works from the pg module for some reason.
db = pg.connect(user="postgres", host=dbHost, dbname=dbName)
db = pg.connect(user=dbUser, host=dbHost, dbname=dbName, port=dbPort, passwd=dbPassword)
sX = sY = sZ = 0
pol = 2 # number of polarizations
for cline in getCoordLines(sys.argv[1]):
if stationname == 'CS002': print cline
(name,X,Y,Z,P,Q,R,rcuX,rcuY) = cline.strip().split(',')
pol = 2 # number of polarizations
for cline in get_coord_lines(sys.argv[1]):
if stationname == 'CS002':
print cline
(name, X, Y, Z, P, Q, R, rcuX, rcuY) = cline.strip().split(',')
# set object type (LBA, HBA, HBA0 or HBA1)
objecttype = name.strip()
print objecttype,
if objecttype == 'CLBA' or objecttype == 'CHBA0' or objecttype == 'CHBA1' or objecttype == 'CHBA':
number = -1
if(objecttype == 'CLBA' or objecttype == 'CHBA0' or
objecttype == 'CHBA1' or objecttype == 'CHBA'):
number = -1
# make sure the object exists
db.query("select * from add_object('%s','%s',%s)" % ( stationname, objecttype, number))
db.query("select * from add_object('%s','%s',%s)" % (stationname, objecttype, number))
# add the coord.
db.query("select * from add_ref_coord('%s','%s',%s,%s,%s,%s,%s,%s,%s,'%s','%s','%s','%s','%s','%s','%s','%s','%s','%s')" %\
( stationname, objecttype, number,\
X, Y, Z, sX, sY, sZ,\
refSys, refFrame, method, date,\
db.query("select * from add_ref_coord('%s','%s',%s,%s,%s,%s,%s,%s,%s,'%s','%s','%s','%s','%s','%s','%s','%s','%s','%s')" %
(stationname, objecttype, number,
X, Y, Z, sX, sY, sZ,
refSys, refFrame, method, date,
pers1, pers2, pers3, absRef, derived, comment))
continue # next line
continue # next line
antType = name[:1]
if antType == 'L':
objecttype = 'LBA'
......@@ -129,25 +170,26 @@ if __name__ == '__main__':
else: # remote station or internation station one hba filed
objecttype = 'HBA'
else:
print '??',name,
print '??', name,
sys.stdout.flush()
# add RCU X coordinates
number = int(name[1:]) * pol
#print objecttype, number
# print objecttype, number
# make sure the object exists
db.query("select * from add_object('%s','%s',%s)" % ( stationname, objecttype, number))
db.query("select * from add_object('%s','%s',%s)" % (stationname, objecttype, number))
# add the coord.
db.query("select * from add_ref_coord('%s','%s',%s,%s,%s,%s,%s,%s,%s,'%s','%s','%s','%s','%s','%s','%s','%s','%s','%s')" %\
( stationname, objecttype, number, X, Y, Z, sX, sY, sZ, refSys, refFrame, method, date, pers1, pers2, pers3, absRef, derived, comment))
db.query("select * from add_ref_coord('%s','%s',%s,%s,%s,%s,%s,%s,%s,'%s','%s','%s','%s','%s','%s','%s','%s','%s','%s')" %
(stationname, objecttype, number, X, Y, Z, sX, sY, sZ, refSys, refFrame, method,
date, pers1, pers2, pers3, absRef, derived, comment))
# add RCU Y coordinates
number = (int(name[1:]) * pol) + 1
#print objecttype, number
# print objecttype, number
# make sure the object exists
db.query("select * from add_object('%s','%s',%s)" % ( stationname, objecttype, number))
db.query("select * from add_object('%s','%s',%s)" % (stationname, objecttype, number))
# add the coord.
db.query("select * from add_ref_coord('%s','%s',%s,%s,%s,%s,%s,%s,%s,'%s','%s','%s','%s','%s','%s','%s','%s','%s','%s')" %\
( stationname, objecttype, number, X, Y, Z, sX, sY, sZ, refSys, refFrame, method, date, pers1, pers2, pers3, absRef, derived, comment))
db.query("select * from add_ref_coord('%s','%s',%s,%s,%s,%s,%s,%s,%s,'%s','%s','%s','%s','%s','%s','%s','%s','%s','%s')" %
(stationname, objecttype, number, X, Y, Z, sX, sY, sZ, refSys, refFrame, method,
date, pers1, pers2, pers3, absRef, derived, comment))
print ' Done'
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment