Skip to content

Instantly share code, notes, and snippets.

@olzhas23

olzhas23/mongobackup.py

Forked from jasonwyatt/mongobackup.py
Created May 3, 2018 16:18
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save olzhas23/5fe2e29e3df2ae9315b83f235e97a995 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save olzhas23/5fe2e29e3df2ae9315b83f235e97a995 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. @jasonwyatt jasonwyatt created this gist Jan 10, 2013.
    82 changes: 82 additions & 0 deletions mongobackup.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,82 @@
    #!/usr/bin/env python
    import os
    import argparse
    import logging
    import datetime
    import urlparse
    import subprocess

    logging.basicConfig(level=logging.INFO)

    parser = argparse.ArgumentParser(description='Backup Mongolab DBs')
    parser.add_argument('-u', '--url',
    type=str,
    required=False,
    default=None,
    help='Mongo DB URL for Backups')
    parser.add_argument('-o', '--output_dir',
    type=str,
    required=False,
    default='./',
    help='Output directory for the backup.')

    def backup(args):
    today = datetime.datetime.now()

    url = args.url

    if url is None:
    logging.info('Fetching MONGOLAB_URI using heroku config:get')

    url = subprocess.check_output([
    'heroku',
    'config:get',
    'MONGOLAB_URI'
    ]).strip()

    url = urlparse.urlparse(url)
    assert url.scheme == 'mongodb', 'URL must be a MongoDB URL'

    netloc = url.netloc
    username = url.username
    password = url.password
    hostname = url.hostname
    port = url.port
    db = url.path[1:]

    output_dir = os.path.abspath(os.path.join(
    os.path.curdir,
    args.output_dir))

    assert os.path.isdir(output_dir), 'Directory %s can\'t be found.' % output_dir

    output_dir = os.path.abspath(os.path.join(output_dir,
    '%s__%s'% ( db, today.strftime('%Y_%m_%d_%H%M%S'))
    ))

    logging.info('Backing up %s from %s to %s' % (db, hostname, output_dir))

    backup_output = subprocess.check_output(
    [
    'mongodump',
    '-host', '%s' % hostname,
    '-u', '%s' % username,
    '-p', '%s' % password,
    '-d', '%s' % db,
    '--port', '%s' % port,
    '-o', '%s' % output_dir
    ])

    logging.info(backup_output)


    if __name__ == '__main__':
    args = parser.parse_args()

    try:
    backup(args)
    except AssertionError, msg:
    logging.error(msg)