mirror of https://github.com/jumpserver/jumpserver
110 lines
2.5 KiB
Python
110 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
from __future__ import unicode_literals
|
|
from six import string_types
|
|
from itertools import chain
|
|
import string
|
|
import logging
|
|
|
|
from django.shortcuts import reverse as dj_reverse
|
|
from django.conf import settings
|
|
from django.core import signing
|
|
from django.utils import timezone
|
|
|
|
|
|
def reverse(viewname, urlconf=None, args=None, kwargs=None, current_app=None, external=False):
|
|
url = dj_reverse(viewname, urlconf=urlconf, args=args, kwargs=kwargs, current_app=current_app)
|
|
|
|
if external:
|
|
url = settings.SITE_URL.strip('/') + url
|
|
return url
|
|
|
|
|
|
def get_object_or_none(model, **kwargs):
|
|
try:
|
|
obj = model.objects.get(**kwargs)
|
|
except model.DoesNotExist:
|
|
obj = None
|
|
return obj
|
|
|
|
|
|
def encrypt(*args, **kwargs):
|
|
return signing.dumps(*args, **kwargs)
|
|
|
|
|
|
def decrypt(*args, **kwargs):
|
|
return signing.loads(*args, **kwargs)
|
|
|
|
|
|
def date_expired_default():
|
|
try:
|
|
years = int(settings.CONFIG.DEFAULT_EXPIRED_YEARS)
|
|
except TypeError:
|
|
years = 70
|
|
|
|
return timezone.now() + timezone.timedelta(days=365 * years)
|
|
|
|
|
|
def combine_seq(s1, s2, callback=None):
|
|
for s in (s1, s2):
|
|
if not hasattr(s, '__iter__'):
|
|
return []
|
|
|
|
seq = chain(s1, s2)
|
|
if callback:
|
|
seq = map(callback, seq)
|
|
return seq
|
|
|
|
|
|
def search_object_attr(obj, value='', attr_list=None, ignore_case=False):
|
|
"""It's provide a method to search a object attribute equal some value
|
|
|
|
If object some attribute equal :param: value, return True else return False
|
|
|
|
class A():
|
|
name = 'admin'
|
|
age = 7
|
|
|
|
:param obj: A object
|
|
:param value: A string match object attribute
|
|
:param attr_list: Only match attribute in attr_list
|
|
:param ignore_case: Ignore case
|
|
:return: Boolean
|
|
"""
|
|
if value == '':
|
|
return True
|
|
|
|
try:
|
|
object_attr = obj.__dict__
|
|
except AttributeError:
|
|
return False
|
|
|
|
if attr_list is not None:
|
|
new_object_attr = {}
|
|
for attr in attr_list:
|
|
new_object_attr[attr] = object_attr.pop(attr)
|
|
object_attr = new_object_attr
|
|
|
|
if ignore_case:
|
|
if not isinstance(value, string_types):
|
|
return False
|
|
|
|
if value.lower() in map(string.lower, map(str, object_attr.values())):
|
|
return True
|
|
else:
|
|
if value in object_attr.values():
|
|
return True
|
|
return False
|
|
|
|
|
|
def get_logger(name=None):
|
|
return logging.getLogger('jumpserver.%s' % name)
|
|
|
|
|
|
def int_seq(seq):
|
|
try:
|
|
return map(int, seq)
|
|
except ValueError:
|
|
return seq
|