reorganize in order to make packaging better
This commit is contained in:
231
nicosman.py
Normal file
231
nicosman.py
Normal file
@ -0,0 +1,231 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# *****************************************************************************
|
||||
#
|
||||
# This program is free software; you can redistribute it and/or modify it under
|
||||
# the terms of the GNU General Public License as published by the Free Software
|
||||
# Foundation; either version 2 of the License, or (at your option) any later
|
||||
# version.
|
||||
#
|
||||
# This program is distributed in the hope that it will be useful, but WITHOUT
|
||||
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
||||
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
||||
# details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License along with
|
||||
# this program; if not, write to the Free Software Foundation, Inc.,
|
||||
# 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
||||
#
|
||||
# Module authors:
|
||||
# Markus Zolliker <markus.zolliker@psi.ch>
|
||||
#
|
||||
# *****************************************************************************
|
||||
|
||||
import os
|
||||
import sys
|
||||
import shutil
|
||||
from glob import glob
|
||||
from os.path import join, abspath, dirname, expanduser, exists, islink
|
||||
from configparser import ConfigParser
|
||||
from servman.base import ServiceManager
|
||||
|
||||
|
||||
ENV_KEYS = {
|
||||
'NICOS_CACHE_PORT',
|
||||
'NICOS_DAEMON_PORT',
|
||||
'FRAPPY_MAIN_PORT',
|
||||
'FRAPPY_STICK_PORT',
|
||||
'FRAPPY_ADDONS_PORT',
|
||||
'SEA_PORT',
|
||||
}
|
||||
|
||||
|
||||
def copy_all(srcdir, dstdir):
|
||||
"""copy all files from srcdir to dstdir"""
|
||||
files = glob(join(srcdir, '*'))
|
||||
for src in files:
|
||||
shutil.copy2(src, dstdir)
|
||||
return files
|
||||
|
||||
|
||||
class NicosManager(ServiceManager):
|
||||
group = 'nicos'
|
||||
services = ('cache', 'daemon', 'poller')
|
||||
|
||||
def do_create(self, ins, *args):
|
||||
self.get_info()
|
||||
if ins == 'all' or ins == 'check':
|
||||
inslist = list(self.info)
|
||||
else:
|
||||
inslist = [ins]
|
||||
for ins_i in inslist:
|
||||
env = self.env[ins_i]
|
||||
base = join(env['NICOS_ROOT'], env['NICOS_PACKAGE'], ins)
|
||||
nicos_conf = join(base, 'nicos.conf')
|
||||
content = {
|
||||
'nicos': {
|
||||
'setup_subdirs': '%s, common, frappy' % ins,
|
||||
'logging_path': '%s/%s' % (env['NICOS_LOG'], ins),
|
||||
'pid_path': '%s/%s' % (env['NICOS_LOG'], ins),
|
||||
},
|
||||
'environment': {
|
||||
key: env[key] for key in env if key in ENV_KEYS
|
||||
}
|
||||
}
|
||||
try:
|
||||
cp = ConfigParser()
|
||||
cp.optionsxform = str
|
||||
cp.read(nicos_conf)
|
||||
if set(cp.sections()) != set(content):
|
||||
oldcontent = None
|
||||
else:
|
||||
oldcontent = {key: cp[key] for key in content}
|
||||
except FileNotFoundError:
|
||||
oldcontent = None
|
||||
if content != oldcontent:
|
||||
cp = ConfigParser()
|
||||
cp.optionxform = str
|
||||
os.makedirs(base, exist_ok=True)
|
||||
for key, sdict in content.items():
|
||||
cp[key] = sdict
|
||||
with open(nicos_conf, 'w') as fd:
|
||||
cp.write(fd)
|
||||
shutil.copyfile(join(base, '..', 'common', 'guiconfig.py'), join(base, 'guiconfig.py'))
|
||||
|
||||
@staticmethod
|
||||
def extra_info(ins):
|
||||
try:
|
||||
datadir = join(os.environ.get('NICOS_DATA', '.'), ins)
|
||||
datadir = join(datadir, os.readlink(join(datadir, 'current')))
|
||||
except FileNotFoundError:
|
||||
return None
|
||||
return 'nicos data: %s' % datadir
|
||||
|
||||
@staticmethod
|
||||
def linked_dir(current_wd):
|
||||
link = join(current_wd, 'data')
|
||||
if islink(link):
|
||||
return join(link, os.readlink(link))
|
||||
if exists(link):
|
||||
raise ValueError('%s is not a symlink' % link)
|
||||
return None
|
||||
|
||||
@staticmethod
|
||||
def handle_linked(target, wd=None):
|
||||
if not target:
|
||||
print('no links from %s' % os.getcwd())
|
||||
return
|
||||
targetdir = abspath(join(target, '..'))
|
||||
analist_file = join(targetdir, 'data_links.txt')
|
||||
linked_from = []
|
||||
if exists(analist_file):
|
||||
with open(analist_file) as f:
|
||||
for line in f:
|
||||
line = line.strip()
|
||||
if line and line != wd:
|
||||
if islink(join(line, 'data')):
|
||||
linked_from.append(line)
|
||||
if wd:
|
||||
linked_from.append(wd)
|
||||
linked_from = '\n'.join(linked_from)
|
||||
if wd:
|
||||
with open(analist_file, 'w') as f:
|
||||
f.write('%s\n' % linked_from)
|
||||
print('links to %s from:\n%s' % (target, linked_from))
|
||||
|
||||
@staticmethod
|
||||
def make_symlink(ins):
|
||||
data = join(os.environ['NICOS_DATA'], ins)
|
||||
target = join(data, os.readlink(join(data, 'current')), 'data')
|
||||
link = join(os.getcwd(), 'data')
|
||||
if islink(link):
|
||||
os.remove(link)
|
||||
elif exists(link):
|
||||
raise ValueError('%s is not a symlink' % link)
|
||||
os.symlink(target, link, True)
|
||||
NicosManager.handle_linked(target, os.getcwd())
|
||||
|
||||
@staticmethod
|
||||
def copy_linked(datadir=None):
|
||||
src = NicosManager.linked_dir(os.getcwd())
|
||||
if src:
|
||||
if not datadir:
|
||||
if src.rsplit('/', 1)[-1] != 'data':
|
||||
raise ValueError('%s is already a copy' % src)
|
||||
datadir = dirname(src)
|
||||
else:
|
||||
if not datadir:
|
||||
raise ValueError('missing data dir')
|
||||
src = join(datadir, 'data')
|
||||
dst = join(os.getcwd(), '%s_%s_data' % tuple(datadir.rsplit('/', 2)[-2:]))
|
||||
os.makedirs(dst, exist_ok=True)
|
||||
n = len(copy_all(src, dst))
|
||||
print('copy %d files to %s' % (n, dst))
|
||||
|
||||
# copy scripts
|
||||
src = join(datadir, 'scripts')
|
||||
dst = join(dst, 'scripts')
|
||||
os.makedirs(dst, exist_ok=True)
|
||||
n = len(copy_all(src, dst))
|
||||
print('copy %d files to %s' % (n, dst))
|
||||
|
||||
@staticmethod
|
||||
def do_link(ins, *args):
|
||||
"""make/show symlink to nicos data in current wd"""
|
||||
if ins is None:
|
||||
NicosManager.handle_linked(NicosManager.linked_dir(os.getcwd()))
|
||||
else:
|
||||
NicosManager.make_symlink(ins)
|
||||
sys.exit(0)
|
||||
|
||||
@staticmethod
|
||||
def do_copy(ins, year_prop, *args):
|
||||
"""copy nicos data to current wd"""
|
||||
if ins is None:
|
||||
NicosManager.copy_linked()
|
||||
else:
|
||||
data = join(os.environ['NICOS_DATA'], ins)
|
||||
if year_prop:
|
||||
year, sep, proposal = year_prop.rpartition('/')
|
||||
if year:
|
||||
src = join(data, year, proposal)
|
||||
else:
|
||||
# get proposal from most recent year
|
||||
src = next(reversed(sorted(glob(join(data, '*', proposal)))))
|
||||
else:
|
||||
src = join(data, os.readlink(join(data, 'current')))
|
||||
NicosManager.copy_linked(src)
|
||||
|
||||
def prepare_start(self, ins):
|
||||
start_dir, env = super().prepare_start(ins)
|
||||
instr = '%s.%s' % (env['NICOS_PACKAGE'], ins)
|
||||
env['INSTRUMENT'] = instr
|
||||
start_dir = env.get('NICOS_START', start_dir)
|
||||
return start_dir, env
|
||||
|
||||
def prepare_client(self, ins):
|
||||
self.check_running(ins, 'daemon')
|
||||
env = self.prepare_start(ins)[1]
|
||||
os.environ.update(env)
|
||||
os.chdir(join(os.environ['NICOS_ROOT'], env['NICOS_PACKAGE']))
|
||||
sys.path.insert(0, os.environ['NICOS_ROOT'])
|
||||
|
||||
def run_client(self, ins, main, app, **kwargs):
|
||||
serverhost = os.environ.get('NICOS_SERVER_HOST', 'localhost')
|
||||
sys.argv[:] = [app, 'guest:guest@%s:%d' % (serverhost, self.info[ins]['daemon'])]
|
||||
sys.exit(main(sys.argv, **kwargs))
|
||||
|
||||
def do_cli(self, ins):
|
||||
self.prepare_client(ins)
|
||||
from nicos.clients.cli import main
|
||||
os.environ['NICOS_HISTORY_FILE'] = expanduser('~/.nicoshistory_%s' % ins)
|
||||
self.run_client(ins, main, 'nicos-client',
|
||||
userpath=expanduser('~/.config/nicos_client_%s' % ins))
|
||||
|
||||
def do_gui(self, ins):
|
||||
self.prepare_client(ins)
|
||||
from nicos.clients.gui.main import main
|
||||
userpath = expanduser('~/.config/nicos_%s' % ins)
|
||||
print('starting nicos gui %s' % ins, userpath)
|
||||
self.run_client(ins, main, 'nicos-gui', userpath=userpath)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user