예제 #1
0
class AuthToken(Base, BaseModel):
    """
    Stores information about possible alerting options
    """
    __tablename__ = 'auth_tokens'

    id = sa.Column(sa.Integer, primary_key=True, nullable=False)
    token = sa.Column(sa.Unicode(40),
                      nullable=False,
                      default=lambda x: UserService.generate_random_string(40))
    owner_id = sa.Column(
        sa.Unicode(30),
        sa.ForeignKey('users.id', onupdate='CASCADE', ondelete='CASCADE'))
    creation_date = sa.Column(sa.DateTime, default=lambda x: datetime.utcnow())
    expires = sa.Column(sa.DateTime)
    description = sa.Column(sa.Unicode, default='')

    @property
    def is_expired(self):
        if self.expires:
            return self.expires < datetime.utcnow()
        else:
            return False

    def __str__(self):
        return '<AuthToken u:%s t:%s...>' % (self.owner_id, self.token[0:10])
예제 #2
0
class AuthToken(BaseModel, Base):
    """
    Auth tokens that can be used to authenticate as specific users
    """
    __tablename__ = 'auth_tokens'

    id = sa.Column(sa.Integer, primary_key=True, nullable=False)
    token = sa.Column(sa.Unicode(40),
                      nullable=False,
                      default=lambda x: UserService.generate_random_string(40))
    owner_id = sa.Column(
        sa.Integer,
        sa.ForeignKey('users.id', onupdate='CASCADE', ondelete='CASCADE'))
예제 #3
0
    def test_generate_random_string(self, db_session):
        rand_str = UserService.generate_random_string()

        assert len(rand_str) == 7
        assert isinstance(rand_str, six.string_types)