restore

#!/usr/bin/env python
import os
import argparse
import requests

from mist.api import config
from mist.api.portal.tasks import restore_backup


def main():
    portal_host = config.PORTAL_URI.split('//')[1]
    argparser = argparse.ArgumentParser(
        description="Restore a Mist backup"
    )

    argparser.add_argument('backup', help="Backup to restore.")
    argparser.add_argument('--db', help="Database backups to restore.")
    argparser.add_argument('--until',
                           help="Keep restoring previous backups until",
                           default=False)
    argparser.add_argument('--portal', help="Portal nane.",
                           default=portal_host)
    args = argparser.parse_args()

    if args.backup.startswith('s3://'):
        # TODO
        return

    if not args.db:
        dbs = ['mongo', 'influx', 'victoria', 'vault']
    else:
        dbs = []
        if 'mongo' in args.db:
            dbs.append('mongo')
        if 'influx' in args.db:
            dbs.append('influx')
        if 'victoria' in args.db:
            dbs.append('victoria')
        if 'vault' in args.db:
            dbs.append('vault')

    return restore_backup(
        args.backup, portal=args.portal, until=args.until, databases=dbs)


if __name__ == '__main__':
    main()