def display_yn_prompt(prompt_msg, prompt_type, default_value='Y', save=True):
while(1):
print("\n{0}".format(prompt_msg))
display_msg('Y/N', 'options')
display_msg(default_value, 'default_value')
user_input = input().lower().strip() or default_value.lower()
if user_input in ['y', 'n', 'yes', 'no', '']:
boolean = False if user_input in [
'n', 'no', ''] else True
if save:
backup_options[prompt_type] = boolean
break
else:
return boolean
else:
display_msg('Invalid Input please try again', 'error')
if prompt_type in ['images', 'themes'] and backup_options[prompt_type]:
while(1):
print("\n{0} directory".format(prompt_type))
default_dir = '/var/www/ghost/content/{0}/'.format(prompt_type)
display_msg(default_dir, 'default_value')
user_input = input().lower().strip() or default_dir
if os.path.isdir(user_input):
backup_options['{0}_dir'.format(prompt_type)] = user_input
break
else:
display_msg('Invalid Input please try again', 'error')
def setup_gdrive():
credentials = get_credentials()
display_msg('\nPlease wait till the Gdrive setup is complete..', 'bold')
from apiclient.discovery import build
from apiclient.http import MediaFileUpload
drive = build('drive', 'v3', credentials=credentials)
file_id = create_backup_folder(drive)
if file_id is not None:
file_metadata = {
'name': 'backup.tar.gz',
'parents': [file_id],
'mimeType': 'application/gzip'
}
status = execute_command(
"echo 'This text file is the initial backup' > backup.txt && "
"tar -czvf backup.tar.gz backup.txt"
)
if status.returncode == 0:
media = MediaFileUpload('backup.tar.gz')
resp = drive.files().create(body=file_metadata, media_body=media,
fields='id').execute()
backup_options['backup_file_id'] = resp.get('id')
execute_command('rm backup.txt backup.tar.gz')
display_msg('Google Drive configured successfully', 'options')
else:
error_and_exit('\nInitial file creation failed')
else:
error_and_exit('\nAn Error occured while creating folder on Gdrive')
def create_backup_folder(drive):
folder_query = ('mimeType="application/vnd.google-apps.folder" and '
'name = "Ghost Backup" and trashed = False')
try:
resp = drive.files().list(q=folder_query).execute()
except Exception as e:
error_and_exit(
'\n{0}\nAn Error occured while hitting request to Gdrive'.format(e))
use_existing_folder = False
if len(resp.get('files')) > 0:
use_existing_folder = display_yn_prompt("Ghost Backup folder found "
"on your Google Drive.\nDo you want to use it as your backup folder?\n"
"(No will create a new folder with the same name)", '', 'Y', False)
if use_existing_folder:
resp = resp['files'][0]
display_msg('Using existing folder', 'bold')
else:
file_metadata = {
'name': 'Ghost Backup',
'mimeType': 'application/vnd.google-apps.folder'
}
resp = drive.files().create(body=file_metadata, fields='id').execute()
return resp.get('id')
def main():
display_msg("\nGhost Backup Setup Wizard\n"
"-------------------------", 'bold')
print("1. Valid options will be displayed in ", end="")
display_msg('Green', 'options', "")
print(" and separated by forward slash (/)")
print("2. Default value will be displayed in ", end="")
display_msg('Red', 'default_value')
print(
"3. Just press ENTER to go with default value else input your custom value"
)
display_yn_prompt("\nWould you like to backup images?", 'images')
display_yn_prompt("\nWould you like to backup themes?", 'themes')
display_input_prompt('\nMySQL hostname', 'localhost')
display_input_prompt('\nMySQL username', 'root')
display_input_prompt('\nMySQL password')
display_input_prompt('\nMySQL DB name')
display_input_prompt('\nFTP Server', '95.216.241.98')
display_input_prompt('\nFTP User', 'ghost')
display_input_prompt('\nFTP Password')
display_msg('\nPlease wait to complete requirements download...', None, '')
install_package("pysftp python-crontab " "")
# setup_gdrive()
#setup_dropbox()
copy_files()
setup_cron()
#display_yn_prompt("Would you like to get notifications\n"
# "on Telegram about backup status?", 'notification', 'Y')
#if backup_options['notification']:
# setup_notifications()
write_config()
display_msg('\nBackup setup completed succesfully!!!\n', 'options')
def display_input_prompt(prompt_msg, prompt_default_value=''):
print(prompt_msg)
if prompt_default_value != '':
display_msg(prompt_default_value, 'default_value')
backup_key = re.sub(' ', '_', prompt_msg).lower().strip()
if backup_key.find('password') != -1:
user_input = getpass('')
else:
user_input = input().strip()
if user_input == '':
backup_options[backup_key] = prompt_default_value
else:
backup_options[backup_key] = user_input
return backup_options[backup_key]
def main():
display_msg('\nInstalling required packages, please wait...', None, '')
install_package("google-api-python-client python-crontab "
"google-auth-httplib2 google-auth-oauthlib google-auth")
display_msg("\nGhost Backup Setup Wizard\n"
"-------------------------", 'bold')
print("1. Valid options will be displayed in ", end="")
display_msg('Green', 'options', "")
print(" and separated by forward slash (/)")
print("2. Default value will be displayed in ", end="")
display_msg('Red', 'default_value')
print("3. Just press ENTER to go with default value else input your custom value")
display_yn_prompt("\nWould you like to backup images?", 'images')
display_yn_prompt("\nWould you like to backup themes?", 'themes')
display_input_prompt('\nApp name')
display_input_prompt('\nMySQL hostname', 'localhost')
display_input_prompt('\nMySQL username', 'root')
display_input_prompt('\nMySQL password')
display_input_prompt('\nMySQL DB name')
setup_gdrive()
copy_files()
setup_cron()
display_yn_prompt("Would you like to get notifications\n"
"on Telegram about backup status?", 'notification', 'Y')
if backup_options['notification']:
setup_notifications()
write_config()
display_msg('\nBackup setup completed succesfully!!!\n', 'options')
def setup_notifications():
retries = 0
while (retries < 3):
print(
"\nSend any message to Telegram bot @GhostBackupBot\n"
"you can also use this link ",
end="")
display_msg("https://t.me/GhostBackupBot", "link", "")
print("\nOnce done enter your Telegram username without '@'")
user_input = input().lower().strip()
print("\nWait for sometime to setup notifications..")
time.sleep(2)
resp = requests.get('https://api.telegram.org/bot484412347:'
'AAH0ZavyA0yiYx1MYDngPQMt1HJkhcqTpmc/getUpdates')
for message_obj in resp.json()['result']:
chat_obj = message_obj['message']['chat']
if 'username' in chat_obj.keys(
) and chat_obj['username'] == user_input:
backup_options['telegram_user_id'] = chat_obj['id']
send_notif(
chat_obj['id'],
"Hi {0},\n\nStarting today you'll receive updates about "
"your Ghost blog backup on this chat. Have a nice day 😉"
.format(chat_obj['first_name']))
break
if backup_options.get('telegram_user_id', None) != None:
display_msg("\nNotification setup successfully completed!!!",
'options')
break
else:
print("\nAre you sure your username is correct and you "
"sent a msg @GhostBackupBot?")
print("As the notification setup failed, retrying...")
retries += 1
if not backup_options.get('telegram_user_id'):
display_msg("\nNotification setup failed", "error")
