def get_attachments(db_path, phone_number):
# Open database
db = sqlite3.connect(db_path)
cursor = db.cursor()
# Tuple with phone number argument
arg_tup = (phone_number,)
# Get messages for phone number
cursor.execute(get_attachments_sql_command(), arg_tup)
# Process results
for attachment in cursor:
filepath = attachment[0]
received_date = process_date(attachment[1], "%Y-%m_%d-%H_%M_%S")
name = attachment[2]
yield filepath, received_date, name
def get_messages(self):
# No phone number provided to use for query
if not self.identifier:
return
# Open database
db = sqlite3.connect(self.dbPath)
cursor = db.cursor()
# Tuple with phone number argument
arg_tup = (self.identifier, )
# Get messages for phone number
cursor.execute(self.get_messages_sql_command(), arg_tup)
# Process results
for msg in cursor:
if not msg[2]:
continue
name = self.myName if msg[0] == 1 else self.otherName
processed_date = process_date(msg[1])
message = msg[2]
yield name + self.SEPARATOR + processed_date + \
self.SEPARATOR + message + "\n"
