Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

MAINT: get_info s3df compatibility #179

Merged
merged 3 commits into from
Apr 10, 2024
Merged

MAINT: get_info s3df compatibility #179

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
0663059
MAINT: rename get_info -> get_info.py and remove executable bit
ZLLentz Apr 10, 2024
a5b5440
MAINT: add wrapper bash script for get_info
ZLLentz Apr 10, 2024
a6558b8
MAINT: get_info needs exec perms
ZLLentz Apr 10, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
232 changes: 17 additions & 215 deletions scripts/get_info
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,215 +1,17 @@
#!/reg/g/pcds/pyps/conda/py36/envs/pcds-3.4.0/bin/python
import argparse
import socket
import os
import sys

parser = argparse.ArgumentParser()
parser.add_argument("--run", help="get last run", action='store_true')
parser.add_argument("--exp", help="get experiment name", action='store_true')
parser.add_argument("--live", help="ongoing?", action='store_true')
parser.add_argument("--ended", help="ended", action='store_true')
parser.add_argument("--hutch", help="get experiment for hutch xxx")
parser.add_argument("--station", help="optional station for hutch with two daqs, e.g. cxi and mfx")
parser.add_argument("--getHutch", help="get hutch (uppercase)", action='store_true')
parser.add_argument("--gethutch", help="get hutch (lowercase)", action='store_true')
parser.add_argument("--getstation", help="get hutch station (for multiple daqs)", action='store_true')
parser.add_argument("--getbase", help="get base daq name (hutch_station if multiple daqs, otherwise hutch)", action='store_true')
parser.add_argument("--getinstrument", help="get instrument (HUTCH_station if multiple daqs, otherwise hutch)", action='store_true')
parser.add_argument("--getcnf", help="get cnf file name)", action='store_true')
parser.add_argument("--files_for_run", help="get xtc files for run")
parser.add_argument("--nfiles_for_run", help="get xtc files for run")
parser.add_argument("--setExp", help="set experiment name")
args = parser.parse_args()

hutches=['tmo','txi','rix','xpp','xcs','mfx','cxi','mec', 'ued', 'det', 'lfe','kfe','tst', 'las', 'hpl']
foundHutch=False
hutch=''

#populate hutch-specific subnets here:
hutch_subnets={'tmo': ['28','132','133','134','135'],
'txi': ['29','136','137','138','139'],
'rix': ['31','144','145','146','147'],
'xpp': ['22','84','85','86','87'],
'xcs': ['25','80','81','82','83'],
'cxi': ['26','68','69','70','71'],
'mfx': ['24','72','73','74','75'],
'mec': ['27','76','77','78','79'],
'ued': ['36'],
'det': ['58', '59'],
'lfe': ['88','89','90','91'],
'kfe': ['92','93','94','95'],
'tst': ['23','148','149','150','151'],
'las': ['35','160','161','162','163'],
'hpl': ['64']}

if args.hutch:
hutch=args.hutch
if hutch in hutches:
hutch=hutch.upper()
else:
for ihutch in hutches:
if hutch.find(ihutch.upper())>=0:
hutch=ihutch.upper()
foundHutch=True
break
if not foundHutch:
print('unknown_hutch')
sys.exit()
else:
hostname=socket.gethostname()
ip=socket.gethostbyname(hostname)
subnet=ip.split('.')[2]
for ihutch in hutches: #use the IP address to match the host to a hutch by subnet
if subnet in hutch_subnets.get(ihutch):
hutch=ihutch.upper()
foundHutch=True
break
if not foundHutch:
for ihutch in hutches:
if hostname.find(ihutch)>=0:
hutch=ihutch.upper()
foundHutch=True
break
if not foundHutch:
if hostname.find('psusr')>=0:
if hostname.find('psusr13')>=0:
hutch='XPP'
elif hostname.find('psusr21')>=0:
hutch='XCS'
elif hostname.find('psusr22')>=0:
hutch='CXI'
elif hostname.find('psusr23')>=0:
hutch='MEC'
elif hostname.find('psusr24')>=0:
hutch='MFX'
if hutch!='':
foundHutch=True
else:
#then check current path
path=os.getcwd()
for ihutch in hutches:
if path.find(ihutch)>=0:
hutch=ihutch.upper()
foundHutch=True
break
if not foundHutch and path.find('xrt')+hostname.find('xrt')>=-1 or path.find('xtod')+hostname.find('xtod')>=-1:
hutch='LFE' #because we have so many names for the same subnet.
foundHutch=True
if not foundHutch:
#then ask.....outside of python
print('unknown_hutch')
sys.exit()
if args.getHutch:
print(hutch.upper())
sys.exit()
if args.gethutch:
print(hutch.lower())
sys.exit()

#if hutch.lower() in ['mfx','cxi']:
if hutch.lower() in ['cxi']:
nstations = 2
if args.station is not None:
station = int(args.station)
else:
hostname=socket.gethostname()
if 'monitor' in hostname:
station = 1
else:
station = 0
daq_base = '{:}_{:}'.format(hutch.lower(),station)
instrument = '{:}:{:}'.format(hutch.upper(),station)
elif hutch.lower() in ['rix']:
station=2
else:
daq_base = hutch.lower()
instrument = hutch.upper()
nstations = 1
if args.station:
station = int(args.station)
else:
station=0

if hutch.lower()!='rix' and station >= nstations:
print("Invalid --station={:} keyword set for hutch {:}".format(hutch))
sys.exit()

if args.getstation:
print(station)
sys.exit()
elif args.getinstrument:
print(instrument)
sys.exit()
elif args.getbase:
print(daq_base)
sys.exit()
elif args.getcnf:
print(daq_base+'.cnf')
sys.exit()

import requests
import logging
ws_url = "https://pswww.slac.stanford.edu/ws/lgbk"
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

if args.exp:
resp = requests.get(ws_url + "/lgbk/ws/activeexperiment_for_instrument_station", {"instrument_name": hutch, "station": station})
exp = resp.json().get("value", {}).get("name")
print(exp)

if args.run:
try:
resp = requests.get(ws_url + "/lgbk/ws/activeexperiment_for_instrument_station", {"instrument_name": hutch, "station": station})
exp = resp.json().get("value", {}).get("name")
rundoc = requests.get(ws_url + "/lgbk/" + exp + "/ws/current_run").json()["value"]
if not rundoc:
#logger.error("Invalid response from server")
print('No runs taken yet')
else:
if args.ended:
if rundoc.get('end_time', None) is not None:
print(int(rundoc['num']))
else:
print(int(rundoc['num'] - 1)) # Really bogus way to determine this; but copying over from previous code.
else:
print(int(rundoc['num']))
if args.live:
if not rundoc.get('end_time', None):
print('live')
else:
print('ended')
except:
logger.exception("No runs?")
print('No runs taken yet')

if args.files_for_run or args.nfiles_for_run:
if args.files_for_run:
run = int(args.files_for_run)
if args.nfiles_for_run:
run = int(args.nfiles_for_run)

if args.setExp:
exp=args.setExp
else:
resp = requests.get(ws_url + "/lgbk/ws/activeexperiment_for_instrument_station", {"instrument_name": hutch, "station": station})
exp = resp.json().get("value", {}).get("name")

currundoc = requests.get(ws_url + "/lgbk/" + exp + "/ws/current_run").json()["value"]
runLast = int(currundoc['num'])
if run > runLast:
print('run %s not taken yet, last run is %s'%(run,runLast))
else:
file_list = requests.get(ws_url + "/lgbk/" + exp + "/ws/" + str(run) + "/files_for_live_mode").json()["value"]
if args.files_for_run:
for tfile in file_list:
print('/reg/d/psdm/'+tfile)
elif args.nfiles_for_run:
#look at files, remove stream 80, only first chunk, return number.
nFiles=0
for tfile in file_list:
tfilename = '/reg/d/psdm/'+tfile
if tfilename.find('c00')>=0 and tfilename.find('-s8')<0:
nFiles=nFiles+1
print('%d %d'%(nFiles,len(file_list)))
#!/bin/bash
HERE="$(realpath "$(dirname "${BASH_SOURCE[0]}")")"
PYTHON_OPTIONS=(
/cds/group/pcds/pyps/conda/py36/envs/pcds-3.4.0/bin/python
/sdf/group/lcls/ds/ana/sw/conda1/inst/envs/ana-4.0.60-py3/bin/python
)

for PYTHON in "${PYTHON_OPTIONS[@]}"
do
if [ -x "${PYTHON}" ]; then
"${PYTHON}" "${HERE}/get_info.py" "${@}"
exit $?
fi
done

echo "No known python executable is available! Tried all of" "${PYTHON_OPTIONS[@]}"
exit 1
Loading