Last active
November 29, 2021 13:50
-
-
Save onekiloparsec/a7df8dd1f2babd9740e8 to your computer and use it in GitHub Desktop.
Quick import-export script to move WhatsApp discussion threads to Slack.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python3 | |
# -*- coding: utf-8 -*- | |
import os | |
import sys | |
import argparse | |
import datetime | |
def main(): | |
console_prefix = "$ " | |
channel_name = "#whatsapp" | |
description = "Transform exported whatsapp discussions into ready-for-import slack.com threads." | |
parser = argparse.ArgumentParser(description=description) | |
parser.add_argument("input", type=argparse.FileType('r'), help="Input filename") | |
parser.add_argument("-c", "--channel", default=channel_name, help="Slack.com channel name, default: "+channel_name) | |
parser.add_argument("-o", "--output", type=argparse.FileType('w'), help="Output filename") | |
# parser.print_help() | |
args = parser.parse_args() | |
# Print description in case of parse success | |
print("\n 🚀 {0}: {1}\n".format(os.path.basename(sys.argv[0]), description)) | |
input_file = args.input | |
output_file = open("Slack Import "+args.input.name, 'w') if args.output is None else args.output | |
print("{0}input filename: '{1}'".format(console_prefix, input_file.name)) | |
print("{0}output filename: '{1}'".format(console_prefix, output_file.name)) | |
print("{0}slack channel name: '{1}'".format(console_prefix, channel_name)) | |
print("{0}Reading input file...".format(console_prefix)) | |
input_lines = input_file.readlines() | |
usernames_mapping = {} | |
# Looping through raw lines to group combine lines | |
output_line = None | |
output_elements = {} | |
with open(output_file.name, 'w') as outfile: | |
for line in input_lines: | |
try: | |
dt = datetime.datetime.strptime(line[:19], "%d/%m/%Y %H:%M:%S") | |
except ValueError: | |
# We cannot find a date, it's a continuation of a line, most probably... | |
output_elements["content"] += "\n"+line.strip() | |
else: | |
if output_elements.get("content", None) is not None: | |
new_line = '{0} {1} @{2} "{3}"'.format(output_elements["date"].timestamp(), channel_name, output_elements["username"], output_elements["content"]) | |
print(new_line) | |
outfile.write(new_line+"\n") | |
output_elements = {} | |
# We can find a date at start of line, it's a new line | |
output_line = line.strip() | |
output_elements["date"] = dt | |
# Make sure to change all double quotes to standard ones | |
for quote in ['"', '‟', '″', '˝', '“']: | |
output_line = output_line.replace(quote, '\"') | |
# Oh, by the way, look for a username. The presence of a username followed by a colon is the only fkag we can use. | |
if line[20:].count(':') > 0: | |
input_username = line[20:].split(':')[0].strip() | |
if input_username not in usernames_mapping.keys(): | |
output_username = input("\n{0}Unknown username '{1}'. Enter corresponding Slack.com username (<Enter>=identical): ".format(console_prefix, input_username)) | |
if len(output_username.strip()) > 0: | |
usernames_mapping[input_username] = output_username.strip() | |
output_username = usernames_mapping.get(input_username, None) | |
if output_username is not None: | |
output_elements["username"] = output_username | |
output_elements["content"] = line[20:].replace(input_username+":", "").strip() | |
# We need this to get the last line... | |
if output_elements.get("content", None) is not None: | |
new_line = '{0} {1} @{2} "{3}"'.format(output_elements["date"].timestamp(), channel_name, output_elements["username"], output_elements["content"]) | |
print(new_line) | |
outfile.write(new_line+"\n") | |
output_elements = {} | |
print("\n 🌖 {0}Done. Enjoy!\n".format(console_prefix)) | |
if __name__ == "__main__": | |
main() | |
Please add details about how to use it. What command line args are to be passed ?
This thing is 5 years old, and unmaintained, sorry. It's up to you to try and fix.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Getting this error:
Traceback (most recent call last): File "whatsapp2slack.py", line 87, in <module> main() File "whatsapp2slack.py", line 47, in main output_elements["content"] += "\n"+line.strip() KeyError: 'content'