Adding code and updated readme

Author: cobusbernard

README.md

@@ -1,11 +1,6 @@
-## My Project
+## Amazon Keyspaces Python Utility Scripts
 
-TODO: Fill this README out!
-
-Be sure to:
-
-* Change the title in this README
-* Edit your repository description on GitHub
+This repo contains utility scripts to help manage users for the [Scalability and familiarity with Amazon Keyspaces](https://aws.amazon.com/getting-started/hands-on/purpose-built-databases/keyspaces/) tutorial. Please see the tutorial for more information.
 
 ## Security
 
@@ -14,4 +9,3 @@ See [CONTRIBUTING](CONTRIBUTING.md#security-issue-notifications) for more inform
 ## License
 
 This library is licensed under the MIT-0 License. See the LICENSE file.
-

create_user.py

@@ -0,0 +1,27 @@
+from entities import User
+from utils import cluster
+
+session = cluster.connect()
+
+
+def create_user(user):
+    print(f"Creating {user.username}...")
+    command = f"INSERT INTO users.users (username, name, email, address, birthdate) VALUES ('{user.username}', '{user.name}', '{user.email}', '{user.address}', '{user.birthdate}') IF NOT EXISTS"
+    r = session.execute(command)
+    results = r.one()
+    applied = results["[applied]"]
+    if not applied:
+        raise Exception(f"User {user.username} already exists")
+    print(f"User {user.username} created successfully!")
+    return user
+
+
+user = User(
+    username="michelle72",
+    name="Thomas Davis",
+    email="jake74@hotmail.com",
+    address="136 Ryan Knolls, East Tiffany, AL 29438",
+    birthdate="1986-07-14",
+)
+
+create_user(user)

delete_user.py

@@ -0,0 +1,15 @@
+from entities import User
+from utils import cluster
+
+session = cluster.connect()
+
+
+def delete_user(username):
+    print(f"Deleting {username}...")
+    command = f"DELETE FROM users.users WHERE username = '{username}'"
+    r = session.execute(command)
+    results = r.one()
+    print(f"User {username} deleted.")
+
+
+delete_user("michelle72")

entities.py

@@ -0,0 +1,10 @@
+class User:
+    def __init__(self, **kwargs):
+        self.username = kwargs["username"]
+        self.name = kwargs["name"]
+        self.email = kwargs["email"]
+        self.address = kwargs["address"]
+        self.birthdate = kwargs["birthdate"]
+
+    def __repr__(self):
+        return "User<{} -- {}>".format(self.username, self.name)

get_user.py

@@ -0,0 +1,19 @@
+from entities import User
+from utils import cluster
+
+session = cluster.connect()
+
+
+def get_user(username):
+    print(f"Retrieving {username}...")
+    command = f"SELECT * FROM users.users WHERE username = '{username}'"
+    r = session.execute(command)
+    results = r.one()
+    if not results:
+        raise Exception(f"User {username} does not exist")
+    user = User(**results)
+    print(user)
+    return user
+
+
+user = get_user("michelle72")

requirements.txt

@@ -0,0 +1 @@
+cassandra-driver==3.23.0

test_connection.py

@@ -0,0 +1,10 @@
+from entities import User
+from utils import cluster
+
+session = cluster.connect()
+r = session.execute("SELECT * FROM users.users LIMIT 5")
+print(
+    f"Connected to Keyspaces table. There are {len(r.current_rows)} rows in the results."
+)
+for row in r.current_rows:
+    print(User(**row))

utils.py

@@ -0,0 +1,26 @@
+import os
+
+from cassandra.cluster import Cluster, ExecutionProfile, EXEC_PROFILE_DEFAULT
+from cassandra import ConsistencyLevel
+from cassandra.query import dict_factory
+from ssl import SSLContext, PROTOCOL_TLSv1, CERT_REQUIRED
+from cassandra.auth import PlainTextAuthProvider
+
+USERNAME = os.environ["CASSANDRA_USERNAME"]
+PASSWORD = os.environ["CASSANDRA_PASSWORD"]
+REGION = os.environ["AWS_REGION"]
+
+ssl_context = SSLContext(PROTOCOL_TLSv1)
+ssl_context.load_verify_locations("./AmazonRootCA1.pem")
+ssl_context.verify_mode = CERT_REQUIRED
+auth_provider = PlainTextAuthProvider(username=USERNAME, password=PASSWORD)
+profile = ExecutionProfile(
+    consistency_level=ConsistencyLevel.LOCAL_QUORUM, row_factory=dict_factory
+)
+cluster = Cluster(
+    [f"cassandra.{REGION}.amazonaws.com"],
+    ssl_context=ssl_context,
+    auth_provider=auth_provider,
+    port=9142,
+    execution_profiles={EXEC_PROFILE_DEFAULT: profile},
+)