mirror of
https://github.com/EmpireProject/Empire
synced 2026-06-08 10:58:16 +00:00
Dakota Nelson
24 lines
564 B
Python
24 lines
564 B
Python
"""
|
|
Connect to the default database at ./data/empire.db.
|
|
"""
|
|
|
|
import sys
|
|
import sqlite3
|
|
|
|
import helpers
|
|
|
|
def connect_to_db():
|
|
try:
|
|
# set the database connectiont to autocommit w/ isolation level
|
|
conn = sqlite3.connect('./data/empire.db', check_same_thread=False)
|
|
conn.text_factory = str
|
|
conn.isolation_level = None
|
|
return conn
|
|
|
|
except Exception:
|
|
print helpers.color("[!] Could not connect to database")
|
|
print helpers.color("[!] Please run database_setup.py")
|
|
sys.exit()
|
|
|
|
db = connect_to_db()
|