add task 11

Author: patrice012

0x0F-python-object_relational_mapping/11-model_state_insert.py

@@ -0,0 +1,90 @@
+#!/usr/bin/python3
+
+"""
+Module contains script that adds the
+State object “Louisiana” to the database hbtn_0e_6_usa
+"""
+
+
+def connect_db(host, port, username, passwd, db_name, **kwargs):
+    """
+    Use SQLAlchemy to connect to
+    MySQL server running on localhost at port 330
+
+    Args:
+        host(str or int): Database host to use
+        post(int): DB port
+        user(str): DB user to use
+        passwd(str): DB user's password
+        db(str): database name to connect to
+
+    Return:
+        session(Session object): session object bind to our database
+    """
+    from sqlalchemy import create_engine
+    from sqlalchemy.orm import sessionmaker
+
+    if kwargs["dialect"] and kwargs["driver"]:
+        dialect = kwargs["dialect"]
+        driver = kwargs["driver"]
+    else:
+        dialect = "mysql"
+        driver = "mysqldb"
+
+    # create engine object
+    arg = f"{dialect}+{driver}://{username}:{passwd}@{host}:{port}/{db_name}"
+    engine = create_engine(
+        arg,
+        pool_pre_ping=True,
+    )
+    # create session object
+    Session = sessionmaker(bind=engine)
+    # bound sesion object to database
+    session = Session()
+    return session
+
+
+def print_data(data):
+    """
+    Helper function used to print data
+    """
+    try:
+        print("{}".format(data.id))
+    except (AttributeError, IndexError) as e:
+        pass
+
+
+def set_state(av):
+    """
+    Get states with name passed as argument using sqlalchemy ORM
+
+    Args:
+        av: list of arguments
+    """
+    from model_state import Base, State
+
+    # get connection credentials
+    host = "localhost"
+    port = 3306
+    user = av[0]
+    passwd = av[1]
+    db = av[2]
+    # search_params = av[3]
+    kwargs = {"dialect": "mysql", "driver": "mysqldb"}
+
+    # connect to database
+    session = connect_db(host, port, user, passwd, db_name=db, **kwargs)
+    # set data
+    louisiana = State(name="Louisiana")
+    session.add(louisiana)
+    # commit transaction into database
+    session.commit()
+    # state = session.query(State).filter(State.name == "Louisiana").first()
+    print_data(louisiana)
+
+
+if __name__ == "__main__":
+    from sys import argv
+
+    av = argv[1:]
+    set_state(av)

0x0F-python-object_relational_mapping/test.sh

@@ -44,6 +44,11 @@ sudo ./10-model_state_my_get.py root root hbtn_0e_6_usa Illinois
 sudo ./10-model_state_my_get.py root root hbtn_0e_6_usa Texas
 
 
+echo ""
+echo "========== Test task 11 ==========="
+sudo ./7-model_state_fetch_all.py root root hbtn_0e_6_usa 
+
+