def set_password(self, password): try: password = escape(unicode(password)) self.pw_hash = generate_password_hash(password) except (TypeError, NameError) as e: log.error(__name__ + ' :: Hash set error - ' + e.message) self.pw_hash = None
def _del(self, **kwargs): if self.conn: try: return self.conn.delete(kwargs['key']) except KeyError as e: log.error('Missing param -> {0}'.format(e.message)) return False else: log.error('No redis connection.') return False
def check_password(self, password): if self.pw_hash: try: password = escape(unicode(password)) return check_password_hash(self.pw_hash, password) except (TypeError, NameError) as e: log.error(__name__ + ' :: Hash check error - ' + e.message) return False else: return False
def create_table(self, obj_name): """ Method for table creation :param name: schema object name :return: boolean indicating status """ if hasattr(schema, obj_name): getattr(schema, obj_name).__table__.create(bind=self.engine) return True else: log.error('Schema object not found for "%s"' % obj_name) return False
def register_user(self): """ Writes the user credentials to the datastore. """ # 1. Only users not already registered # 2. Ensure that the user is unique # 3. Write the user / pass to the db if not self.active: if not get_user(self.name, by_id=False): add_user(self.name, self.pw_hash) log.debug(__name__ + ' :: Added user {0}'.format(self.name)) else: log.error(__name__ + 'Could not add user {0}'.format(self.name)) self.active = True
def write(self): """ Write a model to redis """ dio_r = DataIORedis() dio_r.connect() # Write the model to redis if self.is_valid(): return dio_r.write(key=self.genkey(), value=self.package()) else: log.error('Invalid model -> "{0}"'.format(str(self._model))) return False
def register_user(self): """ Writes the user credentials to the datastore. """ # 1. Only users not already registered # 2. Ensure that the user is unique # 3. Write the user / pass to the db if not self.active: if not get_user(self.name, by_id=False): add_user(self.name, self.pw_hash) log.debug(__name__ + ' :: Added user {0}'. format(self.name)) else: log.error(__name__ + 'Could not add user {0}'. format(self.name)) self.active = True
def delete(self, qry_obj): """ Method to delete rows from database :param qry_obj: object to delete :return: boolean indicating status of action """ if not self.session: log.error('No session') return False try: self.session.delete(qry_obj) self.session.commit() return True except Exception as e: log.error('Failed to delete row "%s": "%s"' % (str(qry_obj), e.message())) return False
def insert(self, obj_name, **kwargs): """ Method to insert rows in database :param name: object to persist :param **kwargs: field values :return: boolean indicating status of action """ if not self.session: log.error('No session') return False try: self.session.add(getattr(schema, obj_name)(**kwargs)) self.session.commit() return True except Exception as e: log.error('Failed to insert row: "%s"' % e.message) return False
def add_model_process(): """ Handles processing of add model """ model_name = request.form['modelName'] if 'modelName' in request.form \ else None model_type = request.form['modelType'] if 'modelType' in request.form \ else None if model_name and model_type: log.info('Added model (type, name) = "%s", "%s" ' % (model_name, model_type)) mysql = DataIOMySQL() mysql.connect() mysql.insert('Model', uid=-1, name=model_name, mtype=model_type, date_create=int(time.time())) else: log.error('Missing model fields in POST data "%s"' % str(request.data)) return render_template('index_anon.html')