mirror of https://github.com/jumpserver/jumpserver
165 lines
4.1 KiB
Python
165 lines
4.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
from __future__ import unicode_literals
|
|
from six import string_types
|
|
from itertools import chain
|
|
import string
|
|
import logging
|
|
import datetime
|
|
|
|
from itsdangerous import TimedJSONWebSignatureSerializer, JSONWebSignatureSerializer, \
|
|
BadSignature, SignatureExpired
|
|
from django.shortcuts import reverse as dj_reverse
|
|
from django.conf import settings
|
|
from django.core import signing
|
|
from django.utils import timezone
|
|
|
|
SECRET_KEY = settings.SECRET_KEY
|
|
|
|
|
|
def reverse(view_name, urlconf=None, args=None, kwargs=None, current_app=None, external=False):
|
|
url = dj_reverse(view_name, 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
|
|
|
|
|
|
class Signer(object):
|
|
def __init__(self, secret_key=SECRET_KEY):
|
|
self.secret_key = secret_key
|
|
|
|
def sign(self, value):
|
|
s = JSONWebSignatureSerializer(self.secret_key)
|
|
return s.dumps(value)
|
|
|
|
def unsign(self, value):
|
|
s = JSONWebSignatureSerializer(self.secret_key)
|
|
try:
|
|
return s.loads(value)
|
|
except BadSignature:
|
|
return None
|
|
|
|
def sign_t(self, value, expires_in=3600):
|
|
s = TimedJSONWebSignatureSerializer(self.secret_key, expires_in=expires_in)
|
|
return s.dumps(value)
|
|
|
|
def unsign_t(self, value):
|
|
s = TimedJSONWebSignatureSerializer(self.secret_key)
|
|
try:
|
|
return s.loads(value)
|
|
except (BadSignature, SignatureExpired):
|
|
return None
|
|
|
|
|
|
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
|
|
|
|
|
|
def timesince(dt, since='', default="just now"):
|
|
"""
|
|
Returns string representing "time since" e.g.
|
|
3 days, 5 hours.
|
|
"""
|
|
|
|
if since is '':
|
|
since = datetime.datetime.utcnow()
|
|
|
|
if since is None:
|
|
return default
|
|
|
|
diff = since - dt
|
|
|
|
periods = (
|
|
(diff.days / 365, "year", "years"),
|
|
(diff.days / 30, "month", "months"),
|
|
(diff.days / 7, "week", "weeks"),
|
|
(diff.days, "day", "days"),
|
|
(diff.seconds / 3600, "hour", "hours"),
|
|
(diff.seconds / 60, "minute", "minutes"),
|
|
(diff.seconds, "second", "seconds"),
|
|
)
|
|
|
|
for period, singular, plural in periods:
|
|
if period:
|
|
return "%d %s" % (period, singular if period == 1 else plural)
|
|
return default
|
|
|
|
|
|
signer = Signer() |