add task 10

Author: patrice012

0x0F-python-object_relational_mapping/10-model_state_my_get.py

@@ -0,0 +1,92 @@
+#!/usr/bin/python3
+
+"""
+Module contains script that prints the State
+object with the name passed as argument
+from 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[0].id))
+    except (AttributeError, IndexError) as e:
+        # AttributeError: 'Query' object has no attribute 'id' raised
+        # when data[0] does'nt have key == id
+        # IndexError: raised when data is empty
+        print("Not found")
+
+
+def get_states(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)
+    # fetch data
+    queryset = session.query(State).filter(State.name == search_params)
+    states = [state for state in queryset]
+    return states
+
+
+if __name__ == "__main__":
+    from sys import argv
+
+    av = argv[1:]
+    data = get_states(av)
+    print_data(data)

0x0F-python-object_relational_mapping/test.sh

@@ -38,6 +38,11 @@ echo ""
 echo "========== Test task 9 ==========="
 sudo ./9-model_state_filter_a.py root root hbtn_0e_6_usa
 
+echo ""
+echo "========== Test task 10 ==========="
+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
+