Beelance/src/app/models/register.py

from models.database import db
import logging
import mysql.connector

logger = logging.getLogger(__name__)


def set_user(username, password, full_name, company, email,
        street_address, city, state, postal_code, country):
    """
    Register a new user in the database
        :param username: The users unique user name
        :param password: The password
        :param full_name: The users full name
        :param company: The company the user represents
        :param email: The users email address
        :param street_address: The street address of the user
        :param city: The city where the user lives
        :param state: The state where the user lives
        :param postal_code: The corresponding postal code
        :param country: The users country
        :type username: str
        :type password: str
        :type full_name: str
        :type company: str
        :type email: str
        :type street_address: str
        :type city: str
        :type state: str
        :type postal_code: str
        :type country: str
    """
    db.connect()
    cursor = db.cursor()
    query = ("INSERT INTO users VALUES (NULL, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, 0, 0)")
    try:
        cursor.execute(query, (username, password, full_name, company, email, street_address,
                               city, state, postal_code, country))
        db.commit()
    except mysql.connector.Error as err:
        logger.error("Failed executing query: %s", err)
        cursor.fetchall()
        exit(1)
    finally:
        cursor.close()
        db.close()