Added importpackagename

app.py

@@ -1,11 +1,9 @@
 import sqlite3
 from flask import Flask, json, request
 from johnnydep import JohnnyDist
+from utils import *
 app = Flask(__name__)
 
-def get_db_connection():
-    conn = sqlite3.connect('/home/kolby/Documents/GitHub/Mining-Diamonds-I-mean-data/diamonds.db')
-    return conn
 
 @app.route('/')
 def hello_world():  # put application's code here
@@ -16,20 +14,7 @@ def hello_world():  # put application's code here
 def package_to_import():
     packageName = request.args.get('packageName')
 
-    importNames = []
-    conn = get_db_connection()
-    cursor = conn.cursor()
-    post = cursor.execute('SELECT * FROM packages WHERE package = ?', [packageName]).fetchone()
-    if post is None:
-        importNames = JohnnyDist(packageName).import_names
-        posts = cursor.execute('INSERT INTO packages (package) VALUES (?);', [packageName]).fetchone()
-        for i in importNames:
-            posts = cursor.execute('INSERT INTO importNames (importName, packageName) VALUES (?, ?);', (i, packageName)).fetchone()
-        conn.commit()
-    else:
-        bob = cursor.execute('SELECT * FROM importNames WHERE packageName = ?', (packageName,)).fetchall()
-        importNames = bob
-    conn.close()
+    importNames = get_import_name(packageName)
     response = app.response_class(
         response=json.dumps(importNames),
         mimetype='application/json'
@@ -38,24 +23,16 @@ def package_to_import():
 
 @app.route('/importname')
 def import_to_package():
-    packageName = request.args.get('importName')
+    importName = request.args.get('importName')
 
-    importNames = []
+    packageNames = []
     conn = get_db_connection()
     cursor = conn.cursor()
-    post = cursor.execute('SELECT * FROM packages WHERE package = ?', [packageName]).fetchone()
-    if post is None:
-        importNames = JohnnyDist(packageName).import_names
-        posts = cursor.execute('INSERT INTO packages (package) VALUES (?);', [packageName]).fetchone()
-        for i in importNames:
-            posts = cursor.execute('INSERT INTO importNames (importName, packageName) VALUES (?, ?);', (i, packageName)).fetchone()
-        conn.commit()
-    else:
-        bob = cursor.execute('SELECT * FROM importNames WHERE packageName = ?', (packageName,)).fetchall()
-        importNames = bob
+    bob = cursor.execute('SELECT * FROM importNames WHERE importName = ?', [importName]).fetchall()
+    packageNames = [i[2] for i in bob]
     conn.close()
     response = app.response_class(
-        response=json.dumps(importNames),
+        response=json.dumps(packageNames),
         mimetype='application/json'
     )
     return response

utils.py

@@ -0,0 +1,23 @@
+import sqlite3
+from johnnydep import JohnnyDist
+
+def get_db_connection():
+    conn = sqlite3.connect('/home/kolby/Documents/GitHub/Mining-Diamonds-I-mean-data/diamonds.db')
+    return conn
+
+def get_import_name(packageName):
+    importNames = []
+    conn = get_db_connection()
+    cursor = conn.cursor()
+    post = cursor.execute('SELECT * FROM packages WHERE package = ?', [packageName]).fetchone()
+    if post is None:
+        importNames = JohnnyDist(packageName).import_names
+        posts = cursor.execute('INSERT INTO packages (package) VALUES (?);', [packageName]).fetchone()
+        for i in importNames:
+            posts = cursor.execute('INSERT INTO importNames (importName, packageName) VALUES (?, ?);', (i, packageName)).fetchone()
+        conn.commit()
+    else:
+        bob = cursor.execute('SELECT * FROM importNames WHERE packageName = ?', (packageName,)).fetchall()
+        importNames = [i[1] for i in bob]
+    conn.close()
+    return importNames