Imported Upstream version 4.3.1
This commit is contained in:
@@ -1,143 +0,0 @@
|
||||
#
|
||||
# Copyright (C) 2016 FreeIPA Contributors see COPYING for license
|
||||
#
|
||||
|
||||
import errno
|
||||
import json
|
||||
import locale
|
||||
import logging
|
||||
import os
|
||||
import time
|
||||
|
||||
import six
|
||||
|
||||
from . import compat
|
||||
from . import schema
|
||||
from ipaclient.plugins.rpcclient import rpcclient
|
||||
from ipalib.constants import USER_CACHE_PATH
|
||||
from ipapython.dnsutil import DNSName
|
||||
|
||||
# pylint: disable=no-name-in-module, import-error
|
||||
if six.PY3:
|
||||
from collections.abc import MutableMapping
|
||||
else:
|
||||
from collections import MutableMapping
|
||||
# pylint: enable=no-name-in-module, import-error
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class ServerInfo(MutableMapping):
|
||||
_DIR = os.path.join(USER_CACHE_PATH, 'ipa', 'servers')
|
||||
|
||||
def __init__(self, api):
|
||||
hostname = DNSName(api.env.server).ToASCII()
|
||||
self._path = os.path.join(self._DIR, hostname)
|
||||
self._force_check = api.env.force_schema_check
|
||||
self._dict = {}
|
||||
|
||||
# copy-paste from ipalib/rpc.py
|
||||
try:
|
||||
self._language = locale.setlocale(
|
||||
locale.LC_MESSAGES, ''
|
||||
).split('.')[0].lower()
|
||||
except locale.Error:
|
||||
self._language = 'en_us'
|
||||
|
||||
self._read()
|
||||
|
||||
def _read(self):
|
||||
try:
|
||||
with open(self._path, 'r') as sc:
|
||||
self._dict = json.load(sc)
|
||||
except Exception as e:
|
||||
if (isinstance(e, EnvironmentError) and
|
||||
e.errno == errno.ENOENT): # pylint: disable=no-member
|
||||
# ignore non-existent file, this happens when the cache was
|
||||
# erased or the server is contacted for the first time
|
||||
pass
|
||||
else:
|
||||
# warn that the file is unreadable, probably corrupted
|
||||
logger.warning('Failed to read server info: %s', e)
|
||||
|
||||
def _write(self):
|
||||
try:
|
||||
try:
|
||||
os.makedirs(self._DIR)
|
||||
except EnvironmentError as e:
|
||||
if e.errno != errno.EEXIST:
|
||||
raise
|
||||
with open(self._path, 'w') as sc:
|
||||
json.dump(self._dict, sc)
|
||||
except EnvironmentError as e:
|
||||
logger.warning('Failed to write server info: %s', e)
|
||||
|
||||
def __getitem__(self, key):
|
||||
return self._dict[key]
|
||||
|
||||
def __setitem__(self, key, value):
|
||||
self._dict[key] = value
|
||||
|
||||
def __delitem__(self, key):
|
||||
del self._dict[key]
|
||||
|
||||
def __iter__(self):
|
||||
return iter(self._dict)
|
||||
|
||||
def __len__(self):
|
||||
return len(self._dict)
|
||||
|
||||
def update_validity(self, ttl=None):
|
||||
if ttl is None:
|
||||
ttl = 3600
|
||||
self['expiration'] = time.time() + ttl
|
||||
self['language'] = self._language
|
||||
self._write()
|
||||
|
||||
def is_valid(self):
|
||||
if self._force_check:
|
||||
return False
|
||||
|
||||
try:
|
||||
expiration = self._dict['expiration']
|
||||
language = self._dict['language']
|
||||
except KeyError:
|
||||
# if any of these is missing consider the entry expired
|
||||
return False
|
||||
|
||||
if expiration < time.time():
|
||||
# validity passed
|
||||
return False
|
||||
|
||||
if language != self._language:
|
||||
# language changed since last check
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
|
||||
def get_package(api):
|
||||
if api.env.in_tree:
|
||||
# pylint: disable=import-error,ipa-forbidden-import
|
||||
from ipaserver import plugins
|
||||
# pylint: enable=import-error,ipa-forbidden-import
|
||||
else:
|
||||
try:
|
||||
plugins = api._remote_plugins
|
||||
except AttributeError:
|
||||
server_info = ServerInfo(api)
|
||||
|
||||
client = rpcclient(api)
|
||||
client.finalize()
|
||||
|
||||
try:
|
||||
plugins = schema.get_package(server_info, client)
|
||||
except schema.NotAvailable:
|
||||
plugins = compat.get_package(server_info, client)
|
||||
finally:
|
||||
if client.isconnected():
|
||||
client.disconnect()
|
||||
|
||||
object.__setattr__(api, '_remote_plugins', plugins)
|
||||
|
||||
return plugins
|
||||
Reference in New Issue
Block a user