Merge branch 'dev'

This commit is contained in:
KnugiHK
2022-05-09 18:31:03 +08:00
4 changed files with 16 additions and 7 deletions

View File

@@ -1 +1 @@
__version__ = "0.8.1"
__version__ = "0.8.2"

View File

@@ -78,6 +78,13 @@ def main():
dest="template",
default=None,
help="Path to custom HTML template")
parser.add_option(
"-e",
"--embedded",
dest="embedded",
default=False,
action='store_true',
help="Embed media into HTML file")
(options, args) = parser.parse_args()
if options.android and options.iphone:
@@ -167,7 +174,7 @@ def main():
messages(db, data)
media(db, data, options.media)
vcard(db, data)
create_html(data, options.output, options.template)
create_html(data, options.output, options.template, options.embedded)
else:
print(
"The message database does not exist. You may specify the path "

View File

@@ -215,7 +215,8 @@ def messages(db, data):
messages.media_caption
FROM messages
LEFT JOIN messages_quotes
ON messages.quoted_row_id = messages_quotes._id;""")
ON messages.quoted_row_id = messages_quotes._id
WHERE messages.key_remote_jid <> '-1';""")
i = 0
content = c.fetchone()
while content is not None:
@@ -414,12 +415,13 @@ def vcard(db, data):
if not os.path.isdir(base):
Path(base).mkdir(parents=True, exist_ok=True)
for index, row in enumerate(rows):
file_name = "".join(x for x in row[3] if x.isalnum())
media_name = row[3] if row[3] else ""
file_name = "".join(x for x in media_name if x.isalnum())
file_path = f"{base}/{file_name}.vcf"
if not os.path.isfile(file_path):
with open(file_path, "w", encoding="utf-8") as f:
f.write(row[2])
data[row[1]]["messages"][row[0]]["data"] = row[3] + \
data[row[1]]["messages"][row[0]]["data"] = media_name + \
"The vCard file cannot be displayed here, " \
f"however it should be located at {file_path}"
data[row[1]]["messages"][row[0]]["mime"] = "text/x-vcard"
@@ -427,7 +429,7 @@ def vcard(db, data):
print(f"Gathering vCards...({index + 1}/{total_row_number})", end="\r")
def create_html(data, output_folder, template=None):
def create_html(data, output_folder, template=None, embedded=False):
if template is None:
template_dir = os.path.dirname(__file__)
template_file = "whatsapp.html"

View File

@@ -230,7 +230,7 @@ def vcard(db, data):
print(f"Gathering vCards...({index + 1}/{total_row_number})", end="\r")
def create_html(data, output_folder, template=None):
def create_html(data, output_folder, template=None, embedded=False):
if template is None:
template_dir = os.path.dirname(__file__)
template_file = "whatsapp.html"