121 lines
3.3 KiB
Python
121 lines
3.3 KiB
Python
from paramecio2.libraries.db.corefields import PhangoField
|
|
from paramecio2.libraries.db.coreforms import PasswordForm
|
|
from hmac import compare_digest as compare_hash
|
|
import crypt
|
|
|
|
class PasswordField(PhangoField):
|
|
"""Field for check and save passwords"""
|
|
|
|
def __init__(self, name, size=1024, required=False):
|
|
|
|
super(PasswordField, self).__init__(name, size, required)
|
|
self.protected=True
|
|
self.name_form=PasswordForm
|
|
self.default_value=''
|
|
self.encrypt_password=True
|
|
|
|
def check(self, value):
|
|
|
|
self.txt_error=''
|
|
self.error=False
|
|
|
|
value.strip()
|
|
|
|
if value=='':
|
|
|
|
if self.model!=None:
|
|
|
|
if self.model.updated==True:
|
|
self.required=False
|
|
self.check_blank=True
|
|
return ""
|
|
else:
|
|
|
|
self.txt_error=self.error_default
|
|
self.error=True
|
|
|
|
else:
|
|
self.txt_error=self.error_default
|
|
self.error=True
|
|
|
|
else:
|
|
|
|
#if crypt.METHOD_SHA512 in crypt.methods:
|
|
|
|
#salt=crypt.mksalt(crypt.METHOD_SHA512)
|
|
if self.encrypt_password:
|
|
value=crypt.crypt(value)
|
|
|
|
"""
|
|
else:
|
|
|
|
self.txt_error="You need the SHA512 method"
|
|
self.error=True
|
|
return ""
|
|
"""
|
|
|
|
return value
|
|
|
|
@staticmethod
|
|
def verify( password, h):
|
|
"""Static method used for verify a password save using PasswordField"""
|
|
#return bcrypt_sha256.verify(password, h)
|
|
return compare_hash(h, crypt.crypt(password, h))
|
|
|
|
# Old function bcrypt
|
|
|
|
"""
|
|
try:
|
|
|
|
from passlib.hash import bcrypt
|
|
from passlib.hash import bcrypt_sha256
|
|
|
|
class PasswordField(PhangoField):
|
|
|
|
def __init__(self, name, size=1024, required=False):
|
|
|
|
super(PasswordField, self).__init__(name, size, required)
|
|
self.protected=True
|
|
self.name_form=PasswordForm
|
|
self.default_value=''
|
|
|
|
def check(self, value):
|
|
|
|
self.txt_error=''
|
|
self.error=False
|
|
|
|
value.strip()
|
|
|
|
if value=='':
|
|
|
|
if self.model!=None:
|
|
|
|
if self.model.updated==True:
|
|
self.required=False
|
|
self.check_blank=True
|
|
return ""
|
|
else:
|
|
|
|
self.txt_error="The field is empty"
|
|
self.error=True
|
|
|
|
else:
|
|
self.txt_error="The field is empty"
|
|
self.error=True
|
|
|
|
else:
|
|
|
|
#if crypt.METHOD_SHA512 in crypt.methods:
|
|
|
|
#value = bcrypt_sha256.encrypt(value)
|
|
value = bcrypt_sha256.hash(value)
|
|
|
|
return value
|
|
|
|
@staticmethod
|
|
def verify( password, h):
|
|
|
|
return bcrypt_sha256.verify(password, h)
|
|
|
|
except:
|
|
"""
|