Difference between revisions of "Database Reader - Code"
From RoboWiki
(Created page with "Return back to project page: Door camera - Jakub Vojtek Python code for the Database Reader: <syntaxhighlight lang=python> impo...") |
|||
Line 4: | Line 4: | ||
<syntaxhighlight lang=python> | <syntaxhighlight lang=python> | ||
import sqlite3 | import sqlite3 | ||
− | |||
− | |||
− | |||
− | |||
def print_data(): | def print_data(): | ||
− | c.execute('SELECT * FROM images') | + | with sqlite3.connect('database/face_images.db') as conn: |
− | + | c = conn.cursor() | |
− | + | c.execute('SELECT * FROM images') | |
− | + | rows = c.fetchall() | |
− | + | for row in rows: | |
+ | print(row) | ||
def delete_data_with_no_faces(): | def delete_data_with_no_faces(): | ||
− | c.execute('DELETE FROM images WHERE detected_faces = 0') | + | with sqlite3.connect('database/face_images.db') as conn: |
+ | c = conn.cursor() | ||
+ | c.execute('DELETE FROM images WHERE detected_faces = 0') | ||
+ | conn.commit() | ||
+ | if __name__ == '__main__': | ||
+ | print("Data before deletion:") | ||
+ | print_data() | ||
+ | |||
+ | delete_data_with_no_faces() | ||
+ | |||
+ | print("Data after deletion:") | ||
+ | print_data() | ||
− | |||
</syntaxhighlight> | </syntaxhighlight> |
Latest revision as of 17:49, 9 June 2024
Return back to project page: Door camera - Jakub Vojtek
Python code for the Database Reader:
import sqlite3
def print_data():
with sqlite3.connect('database/face_images.db') as conn:
c = conn.cursor()
c.execute('SELECT * FROM images')
rows = c.fetchall()
for row in rows:
print(row)
def delete_data_with_no_faces():
with sqlite3.connect('database/face_images.db') as conn:
c = conn.cursor()
c.execute('DELETE FROM images WHERE detected_faces = 0')
conn.commit()
if __name__ == '__main__':
print("Data before deletion:")
print_data()
delete_data_with_no_faces()
print("Data after deletion:")
print_data()