forked from moloch--/RootTheBox
-
Notifications
You must be signed in to change notification settings - Fork 0
/
RegistrationToken.py
67 lines (47 loc) · 1.85 KB
/
RegistrationToken.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# -*- coding: utf-8 -*-
"""
Created on Sep 22, 2012
@author: moloch
Copyright 2012 Root the Box
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""
import binascii
from builtins import str
from os import urandom
from sqlalchemy import Column
from sqlalchemy.types import Boolean, String
from libs.StringCoding import decode, encode
from models import dbsession
from models.BaseModels import DatabaseObject
def gen_token():
return binascii.hexlify(urandom(3))
class RegistrationToken(DatabaseObject):
"""Registration token definition"""
value = Column(String(6), unique=True, nullable=False, default=gen_token)
used = Column(Boolean, nullable=False, default=False)
@classmethod
def all(cls):
"""Returns a list of all objects in the database"""
return dbsession.query(cls).all()
@classmethod
def by_id(cls, _id):
"""Returns a the object with id of _id"""
return dbsession.query(cls).filter_by(id=_id).first()
@classmethod
def count(cls):
"""Returns a list of all objects in the database"""
return dbsession.query(cls).count()
@classmethod
def by_value(cls, _value):
"""Returns a the object with value of value"""
return dbsession.query(cls).filter_by(value=encode(_value)).first()
def getvalue(self):
return decode(self.value)