class ICFServer(object):
"""
Integrated cluster finder server class.
"""
SESSION_PATH = '%s/.sessions' % os.path.dirname(__file__)
VALID_COMMANDS = [
'login', 'logout', 'ls', 'remove', 'download', 'get_detections',
'get_members', 'create_iid', 'delete_iid', 'crawl', 'run_icf',
'plot_lambda', 'list_iids'
]
"""
This code can be used to create a list of available formats:
(not used directly for performance)
import astropy.io.registry as reg
fmt = [ f[1] for f in reg.get_formats() ]
fmt
','.join(fmt)
"""
VALID_FORMATS = 'ascii,ascii.aastex,ascii.basic,ascii.cds,ascii.commented_header,ascii.daophot,ascii.fixed_width,ascii.fixed_width_no_header,ascii.fixed_width_two_line,ascii.ipac,ascii.latex,ascii.no_header,ascii.rdb,ascii.sextractor,ascii.tab,fits,hdf5,votable,aastex,cds,daophot,ipac,latex,rdb'.split(
',')
def __init__(self):
self.passwords = PasswordManager()
self.runner = ICFRunner('xmatch')
self.runner.override = True
self.current_user = None
def set_cookie(self, cookie):
self.current_user = MyCookie.get_cookie(cookie)
def _is_logged_in(self):
return self.current_user is not None
def get_user_dir(self):
"""
Get a data folder for the current user.
Creates folder if it does not exist.
"""
user_dir = '%s/icfhome/%s' % (os.path.dirname(__file__),
self.current_user)
ensure_dir(user_dir)
return user_dir
def validate_path(self, path):
"""
Check if a given path is in the user directory.
"""
user_dir = self.get_user_dir()
full_path = os.path.abspath('%s/%s' % (user_dir, path))
return os.path.commonprefix((user_dir, full_path)) == user_dir
@binary_method
@print_method
def login(self, user, password):
log.info('%s loggin in...', user)
login_result = self.passwords.verify_password(user, password)
if login_result == self.passwords.PASSWORD_OK:
self.current_user = user
cookie = MyCookie.create_cookie(user)
return """Content-Type: text/html
Set-Cookie: id=%s
Ok""" % cookie
elif login_result == self.passwords.PASSWORD_FAILED:
return get_response(401, "Password failed!")
elif login_result == self.passwords.USER_NOT_EXISTS:
return get_response(401, "User does not exist!")
@text_method
@print_method
def logout(self):
if self._is_logged_in():
return get_response(200)
else:
return get_response(401, 'Not logged in')
@text_method
@print_method
def change_password(self, new_password):
if self._is_logged_in():
self.passwords.create_user(self.current_user, new_password)
return get_response(200, 'Ok for user %s' % self.current_user)
else:
return get_response(403, 'Not allowed')
@text_method
@print_method
def create_iid(self, iid, ra, dec):
if self._is_logged_in():
value = self.runner.create_iid(self.current_user, iid, ra, dec)
self.runner.conn.commit()
return get_response(
200,
'Ok for user %s, new iid = %s' % (self.current_user, value))
else:
return get_response(403, 'Unable to create iid')
@text_method
@print_method
def delete_iid(self, iid):
if self._is_logged_in():
value = self.runner.delete_iid(self.current_user, iid)
self.runner.conn.commit()
return get_response(
200,
'Ok for user %s, iid %s deleted' % (self.current_user, value))
else:
return get_response(403, 'Not logged in')
@text_method
@print_method
def ls(self):
if self._is_logged_in():
user_dir = self.get_user_dir()
cut_position = len(user_dir) + 1
files = glob('%s/*' % user_dir)
return get_response(200,
'\n'.join([f[cut_position:] for f in files]))
else:
return get_response(403, 'Not logged in')
@binary_method
@non_print_method
def download(self, filename):
"""
Download file from the server.
"""
if self._is_logged_in():
user_dir = self.get_user_dir()
filepath = '%s/%s' % (user_dir, filename)
if os.path.exists(filepath) and \
self.validate_path(filepath):
print """Content-Type:application/octet-stream; name="%s"
Content-Disposition: attachment; filename="%s"
""" % (filepath, filepath)
# Actual File Content will go hear.
result_file = open(filepath, "rb")
print result_file.read()
# Close opend file
result_file.close()
return None
else:
return get_response(403, "File not found: %s" % filepath)
else:
return get_response(403, 'Not logged in')
@text_method
@print_method
def run_icf(self, iid):
if self._is_logged_in():
self.runner.run_icf(self.current_user, iid)
return get_response(200, 'Cluster finder done')
else:
return get_response(403, 'Not logged in')
@text_method
@print_method
def crawl(self, iid):
if self._is_logged_in():
self.runner.crawl(self.current_user, iid)
return get_response(200)
else:
return get_response(403, 'Not logged in')
@binary_method
@print_method
def get_detections(self, iid, filename=None, format='votable'):
if self._is_logged_in():
if format not in self.VALID_FORMATS:
return get_response(
400, """Invalid format %s!
valid formats are: %s""" % (format, self.VALID_FORMATS))
user_dir = self.get_user_dir()
if filename is None:
out_file = 'detections_%s.%s' % (iid, format)
else:
out_file = filename
self.runner.get_detections(self.current_user, iid,
'%s/%s' % (user_dir, out_file), format)
return self.download(out_file)
else:
return get_response(403, 'Not logged in')
@binary_method
@print_method
def get_members(self, iid, filename=None, format='votable'):
if self._is_logged_in():
if format not in self.VALID_FORMATS:
return get_response(
400, """Invalid format %s!
valid formats are: %s""" % (format, self.VALID_FORMATS))
user_dir = self.get_user_dir()
if filename is None:
out_file = 'members_%s.%s' % (iid, format)
else:
out_file = filename
self.runner.get_members(self.current_user, iid,
'%s/%s' % (user_dir, out_file), format)
return self.download(out_file)
else:
return get_response(403, 'Not logged in')
@binary_method
@print_method
def list_iids(self, filename=None, format='votable'):
if self._is_logged_in():
if format not in self.VALID_FORMATS:
return get_response(
400, """Invalid format %s!
valid formats are: %s""" % (format, self.VALID_FORMATS))
user_dir = self.get_user_dir()
if filename is None:
out_file = 'iids.%s' % format
else:
out_file = filename
self.runner.list_iids(self.current_user,
'%s/%s' % (user_dir, out_file), format)
return self.download(out_file)
else:
return get_response(403, 'Not logged in')
@binary_method
@print_method
def plot_lambda(self, iid, filename=None):
if self._is_logged_in():
user_dir = self.get_user_dir()
if filename is None:
out_file = '%s/plot_%s.png' % (user_dir, iid)
else:
out_file = filename
self.runner.plot_one(self.current_user, iid,
'%s/%s' % (user_dir, out_file))
return self.download(out_file)
else:
return get_response(403, 'Not logged in')
@text_method
@print_method
def remove(self, filename):
if self._is_logged_in():
if self.validate_path(filename):
user_dir = self.get_user_dir()
silentremove('%s/%s' % (user_dir, filename))
return get_response(200, 'File %s deleted' % filename)
else:
return get_response(403, 'No access to file %s' % filename)
else:
return get_response(403, 'Not logged in')
class ICFServer(object):
ROOT = '/icf'
LOGIN = '******'
def __init__(self):
self.passwords = PasswordManager()
self.runner = ICFRunner('xmatch')
self.runner.override = True
self.current_user = None
self.config = Config('icfcherry.conf')
self.iidmanager = IIDmanager(self)
def check_auth(self):
if self.current_user is None:
raise cherrypy.HTTPRedirect(self.LOGIN)
def start(self):
cherrypy.config.update(self.config)
cherrypy.tree.mount(self, '/', config=self.config)
cherrypy.tree.mount(self.iidmanager, '/iid', config=self.config)
cherrypy.engine.start()
cherrypy.engine.block()
def get_basic(self):
return {'username': self.current_user,
'root': self.ROOT}
def default(self):
template = JINJA.get_template('root.template')
return template.render({'username': self.current_user,
'root': self.ROOT})
@cherrypy.expose
def index(self):
print 'Called index'
if self.current_user is None:
raise cherrypy.HTTPRedirect(self.LOGIN)
else:
return self.default()
@cherrypy.expose
def login(self, username=None, password=None):
if username is None or password is None:
raise cherrypy.HTTPRedirect(self.LOGIN)
login_result = self.passwords.verify_password(username, password)
if login_result == self.passwords.PASSWORD_FAILED:
raise cherrypy.HTTPError(401, "Password failed!")
elif login_result == self.passwords.USER_NOT_EXISTS:
raise cherrypy.HTTPError(401, "User does not exist!")
else:
self.current_user = username
cherrypy.session[SESSION_KEY] = cherrypy.request.login = username
raise cherrypy.HTTPRedirect(self.ROOT)
@cherrypy.expose
def logout(self):
sess = cherrypy.session
username = sess.get(SESSION_KEY, None)
sess[SESSION_KEY] = None
if username:
cherrypy.request.login = None
self.current_user = None
raise cherrypy.HTTPRedirect(self.ROOT)
class ICFServer(object):
ROOT = '/icf'
LOGIN = '******'
def __init__(self):
self.passwords = PasswordManager()
self.runner = ICFRunner('xmatch')
self.runner.override = True
self.current_user = None
self.config = Config('icfcherry.conf')
self.iidmanager = IIDmanager(self)
def check_auth(self):
if self.current_user is None:
raise cherrypy.HTTPRedirect(self.LOGIN)
def start(self):
cherrypy.config.update(self.config)
cherrypy.tree.mount(self, '/', config=self.config)
cherrypy.tree.mount(self.iidmanager, '/iid', config=self.config)
cherrypy.engine.start()
cherrypy.engine.block()
def get_basic(self):
return {'username': self.current_user, 'root': self.ROOT}
def default(self):
template = JINJA.get_template('root.template')
return template.render({
'username': self.current_user,
'root': self.ROOT
})
@cherrypy.expose
def index(self):
print 'Called index'
if self.current_user is None:
raise cherrypy.HTTPRedirect(self.LOGIN)
else:
return self.default()
@cherrypy.expose
def login(self, username=None, password=None):
if username is None or password is None:
raise cherrypy.HTTPRedirect(self.LOGIN)
login_result = self.passwords.verify_password(username, password)
if login_result == self.passwords.PASSWORD_FAILED:
raise cherrypy.HTTPError(401, "Password failed!")
elif login_result == self.passwords.USER_NOT_EXISTS:
raise cherrypy.HTTPError(401, "User does not exist!")
else:
self.current_user = username
cherrypy.session[SESSION_KEY] = cherrypy.request.login = username
raise cherrypy.HTTPRedirect(self.ROOT)
@cherrypy.expose
def logout(self):
sess = cherrypy.session
username = sess.get(SESSION_KEY, None)
sess[SESSION_KEY] = None
if username:
cherrypy.request.login = None
self.current_user = None
raise cherrypy.HTTPRedirect(self.ROOT)