calvin.erfmann/WhatsApp-Chat-Exporter
1
0
Fork 0
mirror of https://github.com/KnugiHK/WhatsApp-Chat-Exporter.git synced 2026-06-01 13:59:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactoring --pretty-print-json and --avoid_encoding_json options

Browse Source
This commit is contained in:
KnugiHK
2024-07-13 12:26:54 +08:00
parent 83fefe585b
commit 7bb2fb2420
1 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
Whatsapp_Chat_Exporter/__main__.py
View File

@@ -93,17 +93,19 @@ def main():
const="result.json", const="result.json",
help="Save the result to a single JSON file (default if present: result.json)") help="Save the result to a single JSON file (default if present: result.json)")
parser.add_argument( parser.add_argument(
'--avoidJSONEnsureAscii', '--avoid_encoding_json',
dest='avoid_json_ensure_ascii', dest='avoid_encoding_json',
default=False, default=False,
action='store_true', action='store_true',
help="Don't encode non-ascii chars in the output json files") help="Don't encode non-ascii characters in the output JSON files")
parser.add_argument( parser.add_argument(
'--prettyPrintJson', '--pretty-print-json',
dest='pretty_print_json', dest='pretty_print_json',
default=False, default=None,
action='store_true', nargs='?',
help="Pretty print the output json") const=2,
type=int,
help="Pretty print the output JSON.")
parser.add_argument( parser.add_argument(
'-d', '-d',
'--db', '--db',
@@ -554,7 +556,11 @@ def main():
if not args.json_per_chat: if not args.json_per_chat:
with open(args.json, "w") as f: with open(args.json, "w") as f:
data = json.dumps(data, ensure_ascii=not args.avoid_json_ensure_ascii, indent=2 if args.pretty_print_json else None) data = json.dumps(
data,
ensure_ascii=not args.avoid_encoding_json,
indent=args.pretty_print_json
)
print(f"\nWriting JSON file...({int(len(data)/1024/1024)}MB)") print(f"\nWriting JSON file...({int(len(data)/1024/1024)}MB)")
f.write(data) f.write(data)
else: else:
@@ -569,7 +575,7 @@ def main():
else: else:
contact = jik.replace('+', '') contact = jik.replace('+', '')
with open(f"{args.json}/{contact}.json", "w") as f: with open(f"{args.json}/{contact}.json", "w") as f:
file_content_to_write = json.dumps(data[jik], ensure_ascii=not args.avoid_json_ensure_ascii, indent=2 if args.pretty_print_json else None) file_content_to_write = json.dumps(data[jik], ensure_ascii=not args.avoid_encoding_json, indent=2 if args.pretty_print_json else None)
f.write(file_content_to_write) f.write(file_content_to_write)
print(f"Writing JSON file...({index + 1}/{total})", end="\r") print(f"Writing JSON file...({index + 1}/{total})", end="\r")
print() print()