mirror of https://github.com/zulip/zulip.git
82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
import argparse
|
|
import os
|
|
from typing import Any
|
|
|
|
'''
|
|
Example usage for testing purposes:
|
|
|
|
Move the data:
|
|
rm -Rf ~/hipchat-data
|
|
mkdir ~/hipchat-data
|
|
./manage.py convert_hipchat_data ~/hipchat-31028-2018-08-08_23-23-22.tar --output ~/hipchat-data
|
|
./manage.py import --destroy-rebuild-database hipchat ~/hipchat-data
|
|
|
|
|
|
Test out the realm:
|
|
./tools/run-dev.py
|
|
go to browser and use your dev url
|
|
|
|
spec:
|
|
https://confluence.atlassian.com/hipchatkb/
|
|
exporting-from-hipchat-server-or-data-center-for-data-portability-950821555.html
|
|
'''
|
|
|
|
from django.core.management.base import BaseCommand, CommandError, CommandParser
|
|
|
|
from zerver.data_import.hipchat import do_convert_data
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = """Convert the HipChat data into Zulip data format."""
|
|
|
|
def add_arguments(self, parser: CommandParser) -> None:
|
|
parser.add_argument('hipchat_tar', nargs='+',
|
|
metavar='<hipchat data tarfile>',
|
|
help="tar of HipChat data")
|
|
|
|
parser.add_argument('--output', dest='output_dir',
|
|
help='Directory to write exported data to.')
|
|
|
|
parser.add_argument('--mask', dest='masking_content',
|
|
action="store_true",
|
|
help='Mask the content for privacy during QA.')
|
|
|
|
parser.add_argument('--slim-mode',
|
|
action="store_true",
|
|
help="Default to no public stream subscriptions if no token is available." +
|
|
" See import docs for details.")
|
|
|
|
parser.add_argument('--token', dest='api_token',
|
|
help='API token for the HipChat API for fetching subscribers.')
|
|
|
|
parser.formatter_class = argparse.RawTextHelpFormatter
|
|
|
|
def handle(self, *args: Any, **options: Any) -> None:
|
|
output_dir = options["output_dir"]
|
|
|
|
if output_dir is None:
|
|
raise CommandError("You need to specify --output <output directory>")
|
|
|
|
if os.path.exists(output_dir) and not os.path.isdir(output_dir):
|
|
raise CommandError(output_dir + " is not a directory")
|
|
|
|
os.makedirs(output_dir, exist_ok=True)
|
|
|
|
if os.listdir(output_dir):
|
|
raise CommandError('Output directory should be empty!')
|
|
|
|
output_dir = os.path.realpath(output_dir)
|
|
|
|
for path in options['hipchat_tar']:
|
|
if not os.path.exists(path):
|
|
raise CommandError(f"Tar file not found: '{path}'")
|
|
|
|
print("Converting data ...")
|
|
do_convert_data(
|
|
input_tar_file=path,
|
|
output_dir=output_dir,
|
|
masking_content=options.get('masking_content', False),
|
|
slim_mode=options['slim_mode'],
|
|
api_token=options.get("api_token"),
|
|
)
|