Initial commit

Author: chris-allan

.gitignore

@@ -0,0 +1,108 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+
+#VIM
+*.swp
+*.swo

README.md

@@ -0,0 +1,2 @@
+# OMERO user token
+

appveyor.yml

@@ -0,0 +1,22 @@
+image: ubuntu1804
+
+install:
+  - sudo apt-get update
+  - sudo apt-get install -y python-setuptools python-wheel twine
+
+build: off
+
+build_script:
+  - python setup.py build
+
+#test_script:
+#  - python3 setup.py test
+
+after_test:
+  - python setup.py bdist_wheel bdist_egg
+
+artifacts:
+  - path: dist/*
+
+cache:
+  - ${HOME}/.eggs -> setup.py

omero_user_token/__init__.py

@@ -0,0 +1,71 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# Copyright (c) 2019 Glencoe Software, Inc. All rights reserved.
+#
+# This software is distributed under the terms described by the LICENSE.txt
+# file you can find at the root of the distribution bundle.  If the file is
+# missing please request a copy by contacting [email protected]
+
+import os
+
+import omero
+import omero.all
+
+from configparser import ConfigParser
+
+from omero.rtypes import unwrap
+
+
+def assert_and_get_token_dir():
+    home = os.path.expanduser('~')
+    token_dir = os.path.join(home, '.omero_user_token')
+    if not os.path.exists(token_dir):
+        os.makedirs(token_dir)
+    os.chmod(token_dir, 0700)
+    return token_dir
+
+
+def assert_and_get_token_path():
+    token_dir = assert_and_get_token_dir()
+    return os.path.join(token_dir, 'token')
+
+
+def assert_and_get_config_path():
+    token_dir = assert_and_get_token_dir()
+    return os.path.join(token_dir, 'config')
+
+
+CONFIG = ConfigParser()
+CONFIG.read(assert_and_get_config_path())
+
+if not CONFIG.has_section('server'):
+    CONFIG.add_section('server')
+    CONFIG.set('server', 'host', 'localhost')
+    CONFIG.set('server', 'port', '4064')
+
+
+def setter(server, port, user, password, time_to_idle):
+    client = omero.client(server.encode('utf-8'), port)
+    try:
+        session = client.createSession(user, password)
+        admin_service = session.getAdminService()
+        session_service = session.getSessionService()
+        ec = admin_service.getEventContext()
+        token = '%s@%s:%s' % (unwrap(session_service.createUserSession(
+            0,  # timeToLiveMilliseconds
+            time_to_idle,  # timetoIdleMilliseconds
+            ec.groupName  # defaultGroup
+        ).uuid), server, port)
+        with open(assert_and_get_token_path(), 'w') as token_file:
+            token_file.write(token)
+        return token
+    finally:
+        client.closeSession()
+
+
+def getter():
+    token_path = assert_and_get_token_path()
+    if os.path.exists(token_path):
+        with open(token_path, 'r') as token_file:
+            return token_file.read()

omero_user_token/cli/__init__.py

@@ -0,0 +1,8 @@
+# encoding: utf-8
+#
+# Copyright (c) 2017 Glencoe Software, Inc. All rights reserved.
+#
+# This software is distributed under the terms described by the LICENCE file
+# you can find at the root of the distribution bundle.
+# If the file is missing please request a copy by contacting
+# [email protected].

omero_user_token/cli/omero_user_token.py

@@ -0,0 +1,59 @@
+# encoding: utf-8
+#
+# Copyright (c) 2019 Glencoe Software, Inc. All rights reserved.
+#
+# This software is distributed under the terms described by the LICENCE file
+# you can find at the root of the distribution bundle.
+# If the file is missing please request a copy by contacting
+# [email protected].
+
+import getpass
+
+import click
+
+from .. import CONFIG, getter, setter
+
+
+@click.group()
+def cli():
+    pass
+
+
+@click.command()
+@click.option(
+    '-s', '--server', default=CONFIG.get('server', 'host'), show_default=True,
+    help="OMERO server hostname"
+)
+@click.option(
+    '-p', '--port', default=CONFIG.getint('server', 'port'), type=int,
+    show_default=True,
+    help="OMERO server port"
+)
+@click.option(
+    '-u', '--user', default=getpass.getuser(), show_default=True,
+    help="OMERO username"
+)
+@click.option(
+    '--time_to_idle', default=2**31 - 1, type=int,
+    help="Number of seconds to set the timeToIdle value to; "
+         "defaults to maximum allowed"
+)
+def _set(server, port, user, time_to_idle):
+    password = getpass.getpass("Password: ")
+    token = setter(server, port, user, password, time_to_idle)
+    print('Successfuly set token: %s' % token)
+
+
+@click.command()
+def _get():
+    token = getter()
+    if token is not None:
+        print(token)
+
+
+cli.add_command(_set, name='set')
+cli.add_command(_get, name='get')
+
+
+def main():
+    cli()

omero_user_token/version.py

@@ -0,0 +1 @@
+VERSION = '0.1.0rc6.post6.dev83821103'

setup.py

@@ -0,0 +1,81 @@
+import os
+import sys
+
+from setuptools import setup, find_packages
+from setuptools.command.test import test as TestCommand
+
+import version
+
+# Hack to prevent stupid "TypeError: 'NoneType' object is not callable" error
+# in multiprocessing/util.py _exit_function when running `python
+# setup.py test` or `python setup.py flake8`.  See:
+#  * http://www.eby-sarna.com/pipermail/peak/2010-May/003357.html)
+#  * https://github.com/getsentry/raven-python/blob/master/setup.py
+import multiprocessing
+assert multiprocessing  # silence flake8
+
+
+class PyTest(TestCommand):
+
+    user_options = [('pytest-args=', 'a', 'Arguments to pass to py.test')]
+
+    def initialize_options(self):
+        TestCommand.initialize_options(self)
+        self.pytest_args = []
+
+    def finalize_options(self):
+        TestCommand.finalize_options(self)
+        self.test_args = []
+        self.test_suite = True
+
+    def run_tests(self):
+        import pytest
+        if isinstance(self.pytest_args, str):
+            # pytest requires arguments as a list or tuple even if singular
+            self.pytest_args = [self.pytest_args]
+        errno = pytest.main(self.pytest_args)
+        sys.exit(errno)
+
+
+def read(fname):
+    """
+    Utility function to read the README file.
+    :rtype : String
+    """
+    return open(os.path.join(os.path.dirname(__file__), fname)).read()
+
+
+setup(name='omero-user-token',
+      version=version.getVersion(),
+      description='OMERO user token management system',
+      long_description=read('README.md'),
+      long_description_content_type='text/markdown',
+      classifiers=[],  # Get strings from
+                       # http://pypi.python.org/pypi?%3Aaction=list_classifiers
+      keywords='',
+      author='Glencoe Software, Inc.',
+      author_email='[email protected]',
+      url='https://github.com/glencoesoftware/omero-user-token',
+      license='License :: OSI Approved :: BSD License',
+      packages=find_packages(),
+      zip_safe=True,
+      include_package_data=True,
+      platforms='any',
+      setup_requires=['flake8'],
+      install_requires=[
+          'click==7.0',
+      ],
+      tests_require=[],
+      cmdclass={'test': PyTest},
+      data_files=[(
+          'config', [
+              'message_consumer.cfg.example',
+              'message-consumer.service'
+          ]
+      )],
+      entry_points={
+          'console_scripts': [
+              'omero_user_token = omero_user_token.cli.omero_user_token:main',
+          ]
+      }
+      )