INNER CODE UNIT · Python
do_add_company
infobyte/emploleaks · emploleaks.py:1900
def do_add_company(self, args):
"""Add a new company to the local database."""
if not args.name:
log.error("Company name is required")
return
cur = self.local_conn.cursor()
try:
cur.execute("INSERT INTO companies (company_name) VALUES (?)", (args.name,))
self.local_conn.commit()
log.info(f"Company '{args.name}' added successfully")
except sqlite3.IntegrityError:
log.error("A company with that name already exists")
self.local_conn.rollback()
except sqlite3.Error as error:
log.error(f"Database error: {error}")
self.local_conn.rollback()
finally: