Fixes

cromosoma/usermodel.py

@@ -0,0 +1,134 @@
+#!/usr/bin/python3
+
+from paramecio.cromosoma.webmodel import WebModel
+from paramecio.cromosoma.coreforms import PasswordForm
+from paramecio.citoplasma.i18n import I18n
+from paramecio.citoplasma.httputils import GetPostFiles
+
+class UserModel(WebModel):
+    
+    def __init__(self, name_field_id="id"):
+        
+        super().__init__(name_field_id) 
+        
+        self.password_field='password'
+        self.email_field='email'
+        self.username_field='username'
+        self.yes_repeat_password=True
+    
+    def create_forms(self, arr_fields={}):
+        
+        # Add password_repeat to forms from the model
+        
+        super().create_forms(arr_fields)
+        
+        if self.password_field in arr_fields and self.yes_repeat_password:
+            
+            repeat_password=PasswordForm('repeat_password', '')
+    
+            repeat_password.required=1
+            
+            repeat_password.label=I18n.lang('common', 'repeat_password', 'Repeat Password')
+            
+            self.create_form_after(self.password_field, repeat_password)
+    """
+    def insert(self, dict_values, external_agent=True):
+        
+        if 'password' in dict_values:
+            
+            dict_values['repeat_password']=dict_values.get('repeat_password', '')
+            
+            if dict_values['repeat_password']!=dict_values[self.password_field]:
+                self.fields[self.password_field].error=True
+                self.fields['password'].txt_error=I18n.lang('common', 'error_passwords_no_match', 'Error: passwords doesn\'t match')
+            
+            return super().insert(dict_values, external_agent)
+    """
+    
+    def check_all_fields(self, dict_values, external_agent, yes_update=False, errors_set="insert"):
+        
+        try:
+            
+            fields, values, update_values=super().check_all_fields(dict_values, external_agent, yes_update, errors_set)
+        except: 
+            
+            return False
+        
+        # Check if passwords matches
+        
+        if self.password_field in dict_values:
+            
+            dict_values['repeat_password']=dict_values.get('repeat_password', '')
+            
+            if dict_values['repeat_password']!=dict_values[self.password_field]:
+                
+                if dict_values[self.password_field].strip()!="":
+                    
+                    self.fields[self.password_field].error=True
+                    self.fields[self.password_field].txt_error=I18n.lang('common', 'error_passwords_no_match', 'Error: passwords doesn\'t match')
+                
+                return False
+
+        # Check if exists user with same email or password
+        
+        get_id=0
+        
+        if self.updated:
+            # Need the id
+            GetPostFiles.obtain_get()
+            GetPostFiles.obtain_post()
+            
+            get_id=GetPostFiles.get.get(self.name_field_id, '0')
+            
+            post_id=GetPostFiles.post.get(self.name_field_id, '0')
+            
+            if get_id!='0':
+                get_id=int(get_id)
+            
+            if post_id!='0':
+                get_id=int(post_id)
+            
+            pass
+        
+        sql_id=''
+        
+        original_conditions=self.conditions
+        
+        self.reset_conditions()
+        
+        if self.username_field in dict_values:
+        
+            self.conditions=['WHERE username=%s', [dict_values[self.username_field]]]
+
+        
+        if self.email_field in dict_values:
+        
+            if len(self.conditions[1])>0:
+        
+                self.conditions[0]+=' OR email=%s'
+            else:
+                self.conditions[0]='WHERE email=%s'
+                self.conditions[1]=[]
+        
+            self.conditions[1].append([dict_values[self.email_field]])
+        
+        if get_id>0:
+            self.conditions[0]=' AND '+self.username_field+'=%s'
+            self.conditions[1].append(get_id)
+            
+        
+        if self.select_count()>0:
+            
+            self.fields[self.username_field].error=True
+            self.fields[self.username_field].txt_error=I18n.lang('common', 'error_username_or_password_exists', 'Error: username or email exists in database')
+            
+            return False
+        
+        self.conditions=original_conditions
+
+        return fields, values, update_values
+        
+    
+    
+    
+