|
1 | 1 | from utils import * |
2 | | -from botocore.exceptions import NoCredentialsError |
3 | 2 | import os |
4 | 3 | from dotenv import load_dotenv |
5 | 4 | import os |
6 | | -import time |
7 | | -import schedule |
8 | 5 | import logging |
| 6 | +import schedule |
9 | 7 |
|
10 | 8 | # Load the .env file |
11 | | -load_dotenv(".env") |
| 9 | +load_dotenv(".env" ,override=True) |
12 | 10 |
|
13 | | -# main |
14 | | -if __name__ == "__main__": |
| 11 | +def main(): |
15 | 12 | # # Load configuration from .env file |
16 | | - mode = os.getenv("MODE") |
| 13 | + mode = os.getenv("MODE", 'backup') |
17 | 14 | backup_dir = os.getenv("BACKUP_DIR") |
18 | 15 | output_dir = os.getenv("OUTPUT_DIR") |
| 16 | + |
| 17 | + print("backup_drr", backup_dir) |
| 18 | + |
| 19 | + # logging.info("backup: {backup_dir}") |
| 20 | + |
19 | 21 | # Set up logging |
20 | 22 | logging.basicConfig(level=logging.INFO, |
21 | 23 | format='%(asctime)s - %(levelname)s - %(message)s', |
|
33 | 35 |
|
34 | 36 | if mode == "restore": |
35 | 37 | logging.info("Restore mode ") |
36 | | - S3_ENDPOINT_URL = os.getenv("S3_ENDPOINT") |
37 | | - S3_ACCESS_KEY = os.getenv("S3_ACCESS_KEY") |
38 | | - S3_SECRET_KEY = os.getenv("S3_SECRET_KEY") |
| 38 | + s3 = connect_s3_from_env() |
39 | 39 | S3_BUCKET = os.getenv("S3_BUCKET") |
40 | | - S3_PREFIX = os.getenv("S3_PREFIX") |
41 | 40 | RESTORE_DIR = os.getenv("RESTORE_DIR") |
42 | | - if (not S3_ENDPOINT_URL or not S3_ACCESS_KEY or not S3_SECRET_KEY or not S3_BUCKET): |
43 | | - logging.error("S3 credentials not found") |
44 | | - s3 = connect_s3(S3_ENDPOINT_URL, S3_ACCESS_KEY, S3_SECRET_KEY) |
| 41 | + if (not RESTORE_DIR): |
| 42 | + logging.error("Restore dir not found") |
| 43 | + return |
45 | 44 | download_dir_from_s3( |
46 | 45 | s3, S3_BUCKET, RESTORE_DIR, output_dir) |
47 | 46 | restore_directory(output_dir, backup_dir) |
48 | | - logging.info(f"Restore finished at {get_timestamp()}") |
| 47 | + logging.info(f"Restore finished at {get_time()}") |
| 48 | + |
| 49 | +# main |
| 50 | +if __name__ == "__main__": |
| 51 | + main() |
0 commit comments