39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import sqlite3
|
||
|
|
import sys
|
||
|
|
|
||
|
|
def main(db_path, query):
|
||
|
|
try:
|
||
|
|
sqliteConnection = sqlite3.connect(db_path)
|
||
|
|
cursor = sqliteConnection.cursor()
|
||
|
|
print("Database created and Successfully Connected to SQLite")
|
||
|
|
|
||
|
|
cursor.execute(query)
|
||
|
|
|
||
|
|
if 'select ' in query:
|
||
|
|
for record in cursor.fetchall():
|
||
|
|
print(record)
|
||
|
|
else:
|
||
|
|
sqliteConnection.commit()
|
||
|
|
cursor.close()
|
||
|
|
|
||
|
|
except sqlite3.Error as error:
|
||
|
|
print("Error while connecting to sqlite", error)
|
||
|
|
finally:
|
||
|
|
if sqliteConnection:
|
||
|
|
sqliteConnection.close()
|
||
|
|
print("The SQLite connection is closed")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if len(sys.argv) != 3:
|
||
|
|
print("ERROR: please pass the DB path and the query you want to execute.")
|
||
|
|
print("\nExamples:\n")
|
||
|
|
print(" python3 this_script.py /tmp/some_db_file \'select * from kv where key=\"charmers.openstack-release-version\";\'")
|
||
|
|
print(" python3 this_script.py /tmp/some_db_file \"\"\"select * from kv;\"\"\"")
|
||
|
|
print(" python3 this_script.py /tmp/some_db_file \"\"\"update kv set data='\\\"queens\\\"' where key='charmers.openstack-release-version';\"\"\"")
|
||
|
|
|
||
|
|
else:
|
||
|
|
main(sys.argv[1], sys.argv[2])
|