Skip to content

Commit

Permalink
refactor: code reorganization
Browse files Browse the repository at this point in the history
  • Loading branch information
ergo committed Feb 18, 2018
1 parent 9566ba4 commit aeedf3a
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 4 deletions.
6 changes: 3 additions & 3 deletions ziggurat_foundations/models/services/user.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
import six
import sqlalchemy as sa

from ziggurat_foundations.utils import generate_random_string
from ziggurat_foundations.models.base import get_db_session
from ziggurat_foundations.models.services import BaseService
from ziggurat_foundations.permissions import (
Expand Down Expand Up @@ -253,7 +254,7 @@ def regenerate_security_code(cls, instance):
:param instance:
:return:
"""
instance.security_code = instance.generate_random_string(64)
instance.security_code = cls.generate_random_string(64)

@staticmethod
def generate_random_string(chars=7):
Expand All @@ -262,8 +263,7 @@ def generate_random_string(chars=7):
:param chars:
:return:
"""
return u''.join(random.sample(string.ascii_letters * 2 + string.digits,
chars))
return generate_random_string(chars)

@classmethod
def by_id(cls, user_id, db_session=None):
Expand Down
13 changes: 12 additions & 1 deletion ziggurat_foundations/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,8 @@
"""
Utility functions.
"""

import random
import string

class PlaceholderPasswordChecker(object):
"""Match and return false on check() for '*' (a password hash
Expand Down Expand Up @@ -43,3 +44,13 @@ def __nonzero__(self):


noop = NOOP()


def generate_random_string(chars=7):
"""
:param chars:
:return:
"""
return u''.join(random.sample(string.ascii_letters * 2 + string.digits,
chars))

0 comments on commit aeedf3a

Please sign in to comment.