NewsBlur/utils/backups/backup_mongo.py
2021-01-19 09:40:13 -05:00

40 lines
1.2 KiB
Python

import os
import sys
import shutil
CURRENT_DIR = os.path.dirname(__file__)
NEWSBLUR_DIR = ''.join([CURRENT_DIR, '/../../'])
sys.path.insert(0, NEWSBLUR_DIR)
os.environ['DJANGO_SETTINGS_MODULE'] = 'newsblur_web.settings'
import time
import s3
COLLECTIONS = "classifier_tag classifier_author classifier_feed classifier_title userstories shared_stories category category_site sent_emails social_profile social_subscription social_services statistics user_search feedback"
if False:
COLLECTIONS += " starred_stories"
date = time.strftime('%Y-%m-%d-%H-%M')
collections = COLLECTIONS.split(' ')
db_name = 'newsblur'
dir_name = 'backup_mongo_%s' % date
filename = '%s.tgz' % dir_name
os.mkdir(dir_name)
for collection in collections:
cmd = 'mongodump --db %s --collection %s -o %s' % (db_name, collection, dir_name)
print("Dumping %s: %s" % (collection, cmd))
os.system(cmd)
print("Compressing %s..." % filename)
cmd = 'tar -zcf %s %s' % (filename, dir_name)
os.system(cmd)
print('Uploading %s to S3...' % filename)
try:
s3.save_file_in_s3(filename, name="mongo/%s" % (filename))
except Exception as e:
print(" ****> Exceptions: %s" % e)
shutil.rmtree(dir_name)
os.remove(filename)