Fixed some errors and updated the directions

README.md

@@ -1,15 +1,27 @@
 # backup-mongodb
-To back up to Google Drive  
-This tool is made in Python3 
+Used to back up mongodb to Google Drive in an encrypted 7z file
 
 #Introduction 
-* pip install pymongo
-* pip install bson
-* pip install httplib2
-* pip install --upgrade google-api-python-client 
-* Write the name of the database on line 28
-* Rename the client_secret.json
-* Placing them(mongodb_backup.py and client_secret.json) in the home directory
-
-##Options
-Write username and password of mongodb on line 36-37 if mongodb requires authentication.
+* pip3 install httplib2
+* pip3 install --upgrade google-api-python-client 
+* apt-get install 7zip-full
+
+##Required:
+line 30: DB Name
+line 35: zip password
+line 31/32: DB Username/Password
+
+Enable Google Drive API:
+Use this wizard [https://console.developers.google.com/start/api?id=drive] to create or select a project in the Google Developers Console and automatically turn on the API. Click Continue, then Go to credentials.
+On the Add credentials to your project page, click the Cancel button.
+At the top of the page, select the OAuth consent screen tab. Select an Email address, enter a Product name if not already set, and click the Save button.
+Select the Credentials tab, click the Create credentials button and select OAuth client ID.
+Select the application type Other, enter the name "mongo-backup", and click the Create button.
+Click OK to dismiss the resulting dialog.
+Click the file_download (Download JSON) button to the right of the client ID.
+Move this file to your directory and rename it client_secret.json.
+
+When you run the script for the first time it will give you a url to authorize the script to access your gdrive.
+
+##Options:
+line 28: Folder name in GDrive

mongodb_backup.py

@@ -1,8 +1,6 @@
 #!/usr/bin/env python3
 from __future__ import print_function
 from os.path import join
-from bson.json_util import dumps
-import pymongo
 import httplib2
 import os
 import sys
@@ -14,7 +12,7 @@
 from googleapiclient.http import MediaFileUpload
 from datetime import date
 import glob
-import tarfile
+import shutil
 
 
 try:
@@ -26,30 +24,35 @@
 SCOPES = 'https://www.googleapis.com/auth/drive'
 CLIENT_SECRET_FILE = str(os.path.abspath(os.path.dirname(__file__))) + '/client_secret.json'
 APPLICATION_NAME = 'mongo-backup'
-JSON_DIR = str(os.path.abspath(os.path.dirname(__file__)))+'/data'
+JSON_DIR = str(os.path.abspath(os.path.dirname(__file__)))+'/mongobackuptmp'
 FOLDER = 'mongodb_backup'
-FILE_MIMYTYPE = 'application/json'
-DB_NAME = 'test'  # input databasename crowi's
+FILE_MIMETYPE = 'application/tar'
+DB_NAME = 'DB NAME' 
+DB_USER = 'DB USERNAME'
+DB_PASS = 'DB PASSWORD'
 EXTENSION = '.json'
-BACKUP_EXTENSION = '.tar.gz'
+BACKUP_EXTENSION = '.zip'
+BACKUP_PASSWORD = 'somepassword'
 
 def mongodb_backup(backup_db_dir):
-    client = pymongo.MongoClient(host="127.0.0.1", port=27017)
-    database = client[DB_NAME]
-    # Please write username and password of mongodb if mongodb requires authentication.
-    # authenticated = database.authenticate("username","passwd")
-    # assert authenticated, "Could not authenticate to database!"
-    collections = database.collection_names()
-    for i, collection_name in enumerate(collections):
-        col = getattr(database, collections[i])
-        collection = col.find()
-        jsonpath = collection_name + EXTENSION
-        jsonpath = join(backup_db_dir, jsonpath)
-        with open(jsonpath, 'w') as jsonfile:
-            jsonfile.write(dumps(collection))
-    tar = tarfile.open(os.path.join(backup_db_dir, 'daxiv_'+date.today().isoformat()+BACKUP_EXTENSION), 'w:gz')
-    tar.add(backup_db_dir)
-    tar.close()
+	command = "mongodump -u "+DB_USER+" -p "+DB_PASS+" --db "+DB_NAME+" --out "+backup_db_dir
+
+	print("Executing command: " + command)
+    callreturn = os.system(command);
+    if (callreturn == 0):
+        print('Backup successful')
+    else:
+        print('Something went wrong, mongodump returned: ' + callreturn)
+
+    command = "7z a -t7z -m0=lzma2 -mx=9 -ms=on -mhe=on -p" + BACKUP_PASSWORD + ' \"' + os.path.join(backup_db_dir, 'NSdb_'+date.today().isoformat()+BACKUP_EXTENSION) + "\" " + backup_db_dir 
+
+    print("Executing command: " + command)
+    callreturn = os.system(command);
+    if (callreturn == 0):
+        print('Backup successful')
+    else:
+        print('Something went wrong, 7zip returned: ' + callreturn)
+
 
 def get_credentials():
     home_dir = os.path.expanduser('~')
@@ -72,52 +75,52 @@ def get_credentials():
 
 
 def create_folder(service):
-    print("Create folder: %s" % (FOLDER))
+    print("Creating folder: %s" % (FOLDER))
     folder_metadata = {
         'name': FOLDER,
         'mimeType': 'application/vnd.google-apps.folder'
     }
     folder = service.files().create(body=folder_metadata, fields='id').execute()
-    print("File ID: %s" % folder.get('id'))
     return folder.get('id')
 
 def get_folder(service):
-    response = service.files().list(q="name="+"'"+FOLDER+"'", spaces='drive').execute()
-    if not response:
+    response = service.files().list(q="mimeType='application/vnd.google-apps.folder' and name="+"'"+FOLDER+"'", spaces='drive', fields='files(id)').execute()
+    if not response.get('files',[]):
         return False
     folder = response.get('files', [])[0]
     return folder.get('id')
 
-
-
 def upload_file(service, file_name, folder_id):
-    media_body = MediaFileUpload(file_name, mimetype=FILE_MIMYTYPE, resumable=True)
+    media_body = MediaFileUpload(file_name, mimetype=FILE_MIMETYPE, resumable=True)
     body = {
         'name': os.path.split(file_name)[-1],
-        'mimeType': FILE_MIMYTYPE,
+        'mimeType': FILE_MIMETYPE,
         'parents': [folder_id],
     }
     service.files().create(body=body, media_body=media_body).execute()
 
 def main():
-    mongodb_backup(JSON_DIR)
     credentials = get_credentials()
     http = credentials.authorize(httplib2.Http())
     service = discovery.build('drive', 'v3', http=http)
+    folder_id = get_folder(service)
+    if not folder_id:
+        folder_id = create_folder(service)
+
+    mongodb_backup(JSON_DIR)
     file_path = os.path.join(JSON_DIR, '*' + BACKUP_EXTENSION)
     files = glob.glob(file_path)
     if not files:
         print("No files to upload.")
         sys.exit()
-    folder_id = get_folder(service)
-    if not folder_id:
-        folder_id = create_folder(service)
+
     for file_name in files:
         if os.path.split(file_name)[-1] == 'client_secret.json':
             continue
         print('upload: ' + file_name)
         upload_file(service, file_name, folder_id)
-        os.remove(file_name)
+
+    shutil.rmtree(JSON_DIR)
 
 if __name__ == "__main__":
     main()