Read SQLite database

read_db.py

import sqlite3
import sys

def get_gene_ids(gene_symbol):
    try:
        # estabilish a SQLIte connection from Python
        sqlite_connection = sqlite3.connect('refGene_hg19.db')
        # Better formatting for results
        sqlite_connection.row_factory = lambda cursor, row: row[0]
        # create a cursor object using the connection object
        cursor = sqlite_connection.cursor()
        print('Successfully Connected to SQLite.')
        # define SQLite SELECT statement query
        sqlite_select_query = "SELECT bin from refGene where name2 = (?)"
        # execute the SELECT query by bring the cursor.execute method
        cursor.execute(sqlite_select_query, (gene_symbol,))
        # use cursor.fetchall method to select a row
        return cursor.fetchall()


    except sqlite3.Error as error:
        print("Fail to ?", error)
    finally:
        if(sqlite_connection):
            sqlite_connection.close()
            print("The SQLite connection is closed.")

input_gene_symbol = sys.argv[1]

gene_ids = get_gene_ids(input_gene_symbol)

if len(gene_ids) > 0:
    print("DB has following gene ids: ", *gene_ids)
else:
    print("No results")