Skip to content

GitLab

python-kitgvapi
python-kitgvapi
Commit b14e7ab6 authored by jens.kleineheismann's avatar jens.kleineheismann
Browse files
Options

Usage redesign: introduce KitGvProvider ... 

to provide objects. Removed basicConfig()
Usage/Configuration should be simple AND thread safe now.
parent 9012b335
Hide whitespace changes
Inline Side-by-side
Showing with 264 additions and 230 deletions
README.rst
View file @ b14e7ab6
...@@ -18,28 +18,25 @@ USAGE ...@@ -18,28 +18,25 @@ USAGE
===== =====
:: ::
from kitgvapi import basicConfig, Oe, Group, User from kitgvapi import KitGvProvider
basicConfig('myusername', 'mypassword') gv = KitGvProvider()
gv.auth('myusername', 'mypassword')
iism = Oe('IISM')
iism = gv.getOe('IISM')
for g in iism.get_groups(): for g in iism.get_groups():
print g.name print g.name
iism.create_group('IISM-EM-newgroup', 'Neue Gruppe') iism.create_group('IISM-EM-newgroup', 'Neue Gruppe')
staff = Group('IISM-EM-OU-Staff') staff = gv.getGroup('IISM-EM-OU-Staff')
for u in staff.get_users(effective=True): for u in staff.get_users(effective=True):
print "%s %s (%s)" % (u.firstname, u.lastname, u.samaccount) print "{fn} {ln} ({un})".format(fn=u.firstname, ln=u.lastname, un=u.samaccount)
for g in staff.get_groups(): for g in staff.get_groups():
print "%s (%s)" % (g.name, g.description) print "{gn} ({desc})".format(gn=g.name, desc=g.description)
me = User('zy8373') me = gv.getUser('zy8373')
others = me.oe.get_users() others = me.oe.get_users()
Note:
-----
basicConfig() is not thread safe and is subject to change.
LICENCE LICENCE
======= =======
......
kitgvapi/__init__.py
View file @ b14e7ab6
from .client import Client from .clients import RestClient
from .config import basicConfig
from .models import Group, Oe, User from .models import Group, Oe, User
from .provider import KitGvProvider
kitgvapi/client.py → kitgvapi/clients.py
View file @ b14e7ab6
import json import json
import requests import requests
from .config import CONFIG from .config import DEFAULTS, SYMBOLS
from .exceptions import NotFoundError, MissingConfigError from .exceptions import NotFoundError, MissingConfigError
class ClientConfig(object): class RestClientConfig(object):
def __init__(self): def __init__(self):
self.url_schema = CONFIG['URL_SCHEMA'] self.url_schema = DEFAULTS['URL_SCHEMA']
self.server = CONFIG['SERVER'] self.server = DEFAULTS['SERVER']
self.base_uri = CONFIG['BASE_URI'] self.base_uri = DEFAULTS['BASE_URI']
self.username = CONFIG['USERNAME'] self._username = DEFAULTS['USERNAME']
self.password = CONFIG['PASSWORD'] self._password = DEFAULTS['PASSWORD']
if not self.username: self._credentials = None
self.credentials = None
elif isinstance(self.username, MissingConfigError): def get_username(self):
raise self.username if isinstance(self._username, MissingConfigError):
elif isinstance(self.password, MissingConfigError): raise self._username
raise self.password return self._username
else:
self.credentials = requests.auth.HTTPBasicAuth(self.username, self.password) def set_username(self, username):
self._username = username
self._credentials = None
class Client(object):
def get_password(self):
if isinstance(self._password, MissingConfigError):
raise self._password
return self._password
def set_password(self, password):
self._password = password
self._credentials = None
def get_credentials(self):
if self._credentials is None:
u = self.username
if u:
self._credentials = requests.auth.HTTPBasicAuth(u, self.password)
return self._credentials
username = property(get_username, set_username)
password = property(get_password, set_password)
credentials = property(get_credentials)
class RestClient(object):
def __init__(self, config=None): def __init__(self, config=None):
self.config = config self.config = config
if self.config is None: if self.config is None:
self.config = ClientConfig() self.config = RestClientConfig()
def _get(self, url): def get(self, path):
r = requests.get(url, verify=True, auth=self.config.credentials)
if r.status_code != requests.codes.ok:
r.raise_for_status()
return r
def _post(self, url, data=None, headers=None):
if data is None:
data = {}
if headers is None:
headers = {}
if 'content-type' not in headers:
headers['content-type'] = 'application/json'
r = requests.post(url, verify=True, auth=self.config.credentials, data=data, headers=headers)
if r.status_code != requests.codes.ok:
r.raise_for_status()
return r
def read(self, path):
url = self.config.url_schema + self.config.server + self.config.base_uri + path url = self.config.url_schema + self.config.server + self.config.base_uri + path
try: try:
r = self._get(url) r = requests.get(url, verify=True, auth=self.config.credentials)
r.raise_for_status()
except requests.exceptions.HTTPError as e: except requests.exceptions.HTTPError as e:
if e.response.status_code == 404: if e.response.status_code == 404:
raise NotFoundError("Endpoint returned Not Found") raise NotFoundError("Endpoint returned Not Found")
else: else:
raise e raise e
j = r.json() d = r.json()
return j return d
def write(self, path, data=None): def post(self, path, data=None):
url = self.config.url_schema + self.config.server + self.config.base_uri + path url = self.config.url_schema + self.config.server + self.config.base_uri + path
if data is not None: if data is not None:
data = json.dumps(data) data = json.dumps(data)
r = self._post(url, data) headers = {'content-type': 'application/json'}
r = requests.post(url, verify=True, auth=self.config.credentials, data=data, headers=headers)
r.raise_for_status()
return r
def get_oes(self, pk_only=False): def get_oes(self, pk_only=False):
path = '/oe/list' path = '/oe/list'
l = self.read(path) l = self.get(path)
if pk_only: if pk_only:
l = [d[CONFIG['OE_PK_ATTRIB']] for d in l] l = [d[SYMBOLS['OE_PK_ATTRIB']] for d in l]
return l return l
def get_oe(self, pk): def get_oe(self, pk):
path = '/oe/name/{pk}'.format(pk=pk) path = '/oe/name/{pk}'.format(pk=pk)
return self.read(path) return self.get(path)
def get_oe_groups(self, pk, pk_only=False): def get_oe_groups(self, pk, pk_only=False):
path = '/groups/{pk}'.format(pk=pk) path = '/groups/{pk}'.format(pk=pk)
l = self.read(path) l = self.get(path)
if pk_only: if pk_only:
l = [d[CONFIG['GROUP_PK_ATTRIB']] for d in l] l = [d[SYMBOLS['GROUP_PK_ATTRIB']] for d in l]
return l return l
def get_oe_users(self, pk, pk_only=False): def get_oe_users(self, pk, pk_only=False):
path = '/oe/name/{pk}/identities'.format(pk=pk) path = '/oe/name/{pk}/identities'.format(pk=pk)
l = self.read(path) l = self.get(path)
if pk_only: if pk_only:
l = [d[CONFIG['USER_PK_ATTRIB']] for d in l] l = [d[SYMBOLS['USER_PK_ATTRIB']] for d in l]
return l return l
def get_user(self, pk): def get_user(self, pk):
path = '/identities/get/{pk}'.format(pk=pk) path = '/identities/get/{pk}'.format(pk=pk)
return self.read(path) return self.get(path)
def get_user_groups(self, pk, pk_only=False): def get_user_groups(self, pk, pk_only=False):
path = '/identities/get/{pk}/memberof'.format(pk=pk) path = '/identities/get/{pk}/memberof'.format(pk=pk)
l = self.read(path) l = self.get(path)
if pk_only: if pk_only:
l = [d[CONFIG['GROUP_PK_ATTRIB']] for d in l] l = [d[SYMBOLS['GROUP_PK_ATTRIB']] for d in l]
return l return l
def get_group(self, pk): def get_group(self, pk):
group_pk = pk group_pk = pk
oe_pk = pk.partition('-')[0] oe_pk = pk.partition('-')[0]
path = '/groups/{oe}/{group}'.format(oe=oe_pk, group=group_pk) path = '/groups/{oe}/{group}'.format(oe=oe_pk, group=group_pk)
return self.read(path) return self.get(path)
def get_group_users(self, pk, pk_only=True, effective=False): def get_group_users(self, pk, pk_only=True, effective=False):
if not pk_only: if not pk_only:
raise NotImplemented('Not supported') raise NotImplemented('Not supported')
group_pk = pk group_pk = pk
oe_pk = pk.partition('-')[0] oe_pk = pk.partition('-')[0]
quant = 'members' res = 'members'
if effective: if effective:
quant = 'effective' + quant res = 'effective' + res
path = '/groups/{oe}/{group}/{quant}'.format(oe=oe_pk, group=group_pk, quant=quant) path = '/groups/{oe}/{group}/{res}'.format(oe=oe_pk, group=group_pk, res=res)
return self.read(path) return self.get(path)
def get_group_groups(self, pk, pk_only=True): def get_group_groups(self, pk, pk_only=True):
if not pk_only: if not pk_only:
...@@ -123,21 +131,20 @@ class Client(object): ...@@ -123,21 +131,20 @@ class Client(object):
group_pk = pk group_pk = pk
oe_pk = pk.partition('-')[0] oe_pk = pk.partition('-')[0]
path = '/groups/{oe}/{group}/subgroups'.format(oe=oe_pk, group=group_pk) path = '/groups/{oe}/{group}/subgroups'.format(oe=oe_pk, group=group_pk)
return self.read(path) return self.get(path)
def add_group_group(self, super_pk, sub_pk):
oe_pk = super_pk.partition('-')[0]
path = '/groups/{oe}/{superg}/subgroups/{subg}'.format(oe=oe_pk, superg=super_pk, subg=sub_pk)
self.write(path)
def add_group_user(self, group_pk, user_pk): def add_group_user(self, group_pk, user_pk):
oe_pk = group_pk.partition('-')[0] oe_pk = group_pk.partition('-')[0]
path = '/groups/{oe}/{group}/members/{user}'.format(oe=oe_pk, group=group_pk, user=user_pk) path = '/groups/{oe}/{group}/members/{user}'.format(oe=oe_pk, group=group_pk, user=user_pk)
self.write(path) self.post(path)
def add_group_group(self, super_pk, sub_pk):
oe_pk = super_pk.partition('-')[0]
path = '/groups/{oe}/{superg}/subgroups/{subg}'.format(oe=oe_pk, superg=super_pk, subg=sub_pk)
self.post(path)
def create_group(self, name, description): def create_group(self, name, description):
oe_pk = name.partition('-')[0] oe_pk = name.partition('-')[0]
path = '/groups/{oe}'.format(oe=oe_pk) path = '/groups/{oe}'.format(oe=oe_pk)
data = {'name': name, 'description': description} data = {'name': name, 'description': description}
self.write(path, data=data) self.post(path, data=data)
kitgvapi/config.py
View file @ b14e7ab6
from .exceptions import MissingConfigError from .exceptions import MissingConfigError
CONFIG = { SYMBOLS = {
'GROUP_PK_ATTRIB': 'name', 'GROUP_PK_ATTRIB': 'name',
'OE_PK_ATTRIB': 'kurz', 'OE_PK_ATTRIB': 'kurz',
'USER_PK_ATTRIB': 'samaccount', 'USER_PK_ATTRIB': 'samaccount',
'URL_SCHEMA': 'https://',
'SERVER': 'kit-idm-03.scc.kit.edu',
'BASE_URI': '/itbportal-rest-war/rest',
'USERNAME': MissingConfigError('No username set'),
'PASSWORD': MissingConfigError('No password set'),
} }
def basicConfig(username, password): DEFAULTS = {
CONFIG['USERNAME'] = username 'URL_SCHEMA': 'https://',
CONFIG['PASSWORD'] = password 'SERVER': 'kit-idm-03.scc.kit.edu',
'BASE_URI': '/itbportal-rest-war/rest',
'USERNAME': MissingConfigError('No username set'),
'PASSWORD': MissingConfigError('No password set'),
}
kitgvapi/exceptions.py
View file @ b14e7ab6
class KitGvApiError(Exception): class KitGvApiError(Exception):
pass pass
class MissingConfigError(KitGvApiError): class MissingConfigError(KitGvApiError):
pass pass
class MissingClientError(KitGvApiError): class MissingClientError(KitGvApiError):
pass pass
class MissingPkError(KitGvApiError): class MissingPkError(KitGvApiError):
pass pass
class NotFoundError(KitGvApiError): class NotFoundError(KitGvApiError):
pass pass
kitgvapi/models.py
View file @ b14e7ab6
from .client import Client from .clients import RestClient
from .config import CONFIG from .config import SYMBOLS
from .exceptions import MissingClientError, MissingPkError from .exceptions import MissingClientError, MissingPkError
class Model(object): class Model(object):
def __init__(self, pk=None, client=None, data=None): def __init__(self, pk=None, client=None, data=None):
self.pk = pk self.pk = pk
self.client = client self.client = client
self.is_loaded = False self.is_loaded = False
if self.client is None: if self.client is None:
self.client = Client() self.client = RestClient()
if data is not None: if data is not None:
self._load_data_attribs(data) self._load_data_attribs(data)
self.is_loaded = True self.is_loaded = True
if self.pk is not None: if self.pk is not None:
self.load() self.load()
def _load_data_attribs(self, data=None): def _load_data_attribs(self, data=None):
raise NotImplementedError('Abstract method') raise NotImplementedError('Abstract method')
def load(self, reload=False): def load(self, reload=False):
if self.is_loaded and not reload: if self.is_loaded and not reload:
return return
if self.client is None: if self.client is None:
raise MissingClientError() raise MissingClientError()
if self.pk is None: if self.pk is None:
raise MissingPkError() raise MissingPkError()
self._load_data_attribs() self._load_data_attribs()
self.is_loaded = True self.is_loaded = True
def _save_data_attribs(self): def _save_data_attribs(self):
raise NotImplementedError('Abstract method') raise NotImplementedError('Abstract method')
def save(self): def save(self):
if self.client is None: if self.client is None:
raise MissingClientError() raise MissingClientError()
self._save_data_attribs() self._save_data_attribs()
class Group(Model): class Group(Model):
def _load_data_attribs(self, data=None): def _load_data_attribs(self, data=None):
if data is None: if data is None:
data = self.client.get_group(pk=self.pk) data = self.client.get_group(pk=self.pk)
self.name = data['name'] self.name = data['name']
self.description = data['description'] self.description = data['description']
self.gidNumber = data['gidNumber'] self.gidNumber = data['gidNumber']
self.readOnly = data['readOnly'] self.readOnly = data['readOnly']
self.verteiler = data['verteiler'] self.verteiler = data['verteiler']
self.eMailAddress = data['eMailAddress'] self.eMailAddress = data['eMailAddress']
self.oe_pk = data['oe'] self.oe_pk = data['oe']
self.pk = getattr(self, CONFIG['GROUP_PK_ATTRIB']) self.pk = getattr(self, SYMBOLS['GROUP_PK_ATTRIB'])
def _save_data_attribs(self): def _save_data_attribs(self):
raise NotImplemented('Not supported') raise NotImplementedError('Not supported')
@property @property
def oe(self): def oe(self):
return Oe(client=self.client, pk=self.oe_pk) return Oe(client=self.client, pk=self.oe_pk)
def get_users(self, effective=False): def get_users(self, effective=False):
self.load() self.load()
l = self.client.get_group_users(pk=self.pk, effective=effective) l = self.client.get_group_users(pk=self.pk, effective=effective)
for i in l: for i in l:
y = User(client=self.client, pk=i) y = User(client=self.client, pk=i)
yield y yield y
def add_user(self, pk): def add_user(self, pk):
self.client.add_group_user(group_pk=self.pk, user_pk=pk) self.client.add_group_user(group_pk=self.pk, user_pk=pk)
def remove_user(self, pk): def remove_user(self, pk):
raise NotImplemented('Not yet') raise NotImplementedError('Not yet')
def get_groups(self): def get_groups(self):
self.load() self.load()
l = self.client.get_group_groups(pk=self.pk) l = self.client.get_group_groups(pk=self.pk)
for i in l: for i in l:
y = Group(client=self.client, pk=i) y = Group(client=self.client, pk=i)
yield y yield y
def add_group(self, pk): def add_group(self, pk):
self.client.add_group_group(super_pk=self.pk, sub_pk=pk) self.client.add_group_group(super_pk=self.pk, sub_pk=pk)
def remove_group(self, pk): def remove_group(self, pk):
raise NotImplemented('Not yet') raise NotImplementedError('Not yet')
class Oe(Model): class Oe(Model):
def _load_data_attribs(self, data=None): def _load_data_attribs(self, data=None):
if data is None: if data is None:
data = self.client.get_oe(pk=self.pk) data = self.client.get_oe(pk=self.pk)
self.kurz = data['kurz'] self.kurz = data['kurz']
self.lang = data['lang'] self.lang = data['lang']
self.pk = getattr(self, CONFIG['OE_PK_ATTRIB']) self.pk = getattr(self, SYMBOLS['OE_PK_ATTRIB'])
def _save_data_attribs(self): def _save_data_attribs(self):
raise NotImplemented('Not supported') raise NotImplementedError('Not supported')
def get_groups(self): def get_groups(self):
self.load() self.load()
l = self.client.get_oe_groups(pk=self.pk) l = self.client.get_oe_groups(pk=self.pk)
for d in l: for d in l:
y = Group(client=self.client, data=d) y = Group(client=self.client, data=d)
yield y yield y
def get_users(self): def get_users(self):
self.load() self.load()
l = self.client.get_oe_users(pk=self.pk) l = self.client.get_oe_users(pk=self.pk)
for d in l: for d in l:
y = User(client=self.client, data=d) y = User(client=self.client, data=d)
yield y yield y
def create_group(self, name, description): def create_group(self, name, description):
self.client.create_group(name, description) self.client.create_group(name, description)
def delete_group(self, pk): def delete_group(self, pk):
raise NotImplemented('Not yet') raise NotImplementedError('Not yet')
class User(Model): class User(Model):
def _load_data_attribs(self, data=None): def _load_data_attribs(self, data=None):
if data is None: if data is None:
data = self.client.get_user(pk=self.pk) data = self.client.get_user(pk=self.pk)
self.samaccount = data['samaccount'] self.samaccount = data['samaccount']
self.firstname = data['firstname'] self.firstname = data['firstname']
self.lastname = data['lastname'] self.lastname = data['lastname']
self.oe = Oe(client=self.client, data=data['oe']) self.oe = Oe(client=self.client, data=data['oe'])
self.pk = getattr(self, CONFIG['USER_PK_ATTRIB']) self.pk = getattr(self, SYMBOLS['USER_PK_ATTRIB'])
def _save_data_attribs(self): def _save_data_attr