-
Notifications
You must be signed in to change notification settings - Fork 0
/
upload_log_files.py
154 lines (124 loc) · 7.78 KB
/
upload_log_files.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
import argparse
import os
import re
from core_data_modules.logging import Logger
from storage.google_cloud import google_cloud_utils
from src.lib import PipelineConfiguration
log = Logger(__name__)
def get_file_paths(dir_path):
# search for .gzip (data archive) and .profile (memory profile) files only because os.listdir(dir_path)
# returns all files in the directory
files_list = [file for file in os.listdir(dir_path) if file.endswith((".gzip", ".profile"))]
file_paths = [os.path.join(dir_path, basename) for basename in files_list]
return file_paths
def get_uploaded_file_dates(uploaded_files_list, date_pattern):
dates_match = [re.search(date_pattern, file) for file in uploaded_files_list]
uploaded_file_dates = []
for date_match in dates_match:
if date_match == None:
continue
uploaded_file_dates.append(date_match.group())
return uploaded_file_dates
def get_files_by_date(dir_path, uploaded_file_dates):
file_paths = get_file_paths(dir_path)
files_by_date = {}
if len(file_paths) > 0:
for file in file_paths:
file_date_match = re.search(date_pattern, file)
file_date = file_date_match.group()
if file_date in uploaded_file_dates:
log.info(f" file already uploaded for {file_date}, skipping...")
else:
if file_date not in files_by_date:
files_by_date[file_date] = []
files_by_date[file_date].append(file)
else:
log.info(f" No file found in {dir_path}!, skipping...")
return files_by_date
def delete_old_log_files(dir_path, uploaded_file_dates):
log_file_paths = get_file_paths(dir_path)
files_for_days_that_upload_failed = {}
most_recent_file_path = None
if len(log_file_paths) > 0:
most_recent_file_path = max(log_file_paths, key=os.path.getmtime)
for file_path in log_file_paths:
file_date_match = re.search(date_pattern, file_path)
file_date = file_date_match.group()
# Create a list of files for days that failed to upload
if file_date in uploaded_file_dates:
if file_path == most_recent_file_path:
log.info(f"Retaining latest modified file {file_path} for quick retrieval")
continue
log.warning(f"Deleting {file_path} because files for {file_date} already uploaded to cloud")
os.remove(os.path.join(dir_path, file_path))
# Delete files for days that have a file uploaded in g-cloud
else:
log.debug(f'Files for {file_date} not yet uploaded to cloud, '
f'will delete other files and retain the latest modified file for upload')
if file_date not in files_for_days_that_upload_failed:
files_for_days_that_upload_failed[file_date] = []
files_for_days_that_upload_failed[file_date].append(file_path)
# Check for latest modified file path for each day that failed to upload
# Delete other files for that date
for file_date in files_for_days_that_upload_failed:
most_recent_file_path = max(files_for_days_that_upload_failed[file_date], key=os.path.getmtime)
for file_path in files_for_days_that_upload_failed[file_date]:
if file_path == most_recent_file_path:
log.debug(f"Retaining {file_path}")
continue
log.warning(f"Deleting old file {file_path} for {file_date}")
os.remove(os.path.join(dir_path, file_path))
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Uploads pipeline log files to g-cloud")
parser.add_argument("user", help="User launching this program")
parser.add_argument("google_cloud_credentials_file_path", metavar="google-cloud-credentials-file-path",
help="Path to a Google Cloud service account credentials file to use to access the "
"credentials bucket")
parser.add_argument("pipeline_configuration_file_path", metavar="pipeline-configuration-file-path",
help="Path to the pipeline configuration json file")
parser.add_argument("memory_profile_dir_path", metavar="memory-profile-dir-path",
help="Path to the memory profile log directory with file to upload")
parser.add_argument("data_archive_dir_path", metavar="data-archive-dir-path",
help="Path to the data archive directory with file to upload")
args = parser.parse_args()
user = args.user
google_cloud_credentials_file_path = args.google_cloud_credentials_file_path
pipeline_configuration_file_path = args.pipeline_configuration_file_path
memory_profile_dir_path = args.memory_profile_dir_path
data_archive_dir_path = args.data_archive_dir_path
date_pattern = r'\d{4}-\d{2}-\d{2}'
log.info("Loading Pipeline Configuration File...")
with open(pipeline_configuration_file_path) as f:
pipeline_configuration = PipelineConfiguration.from_configuration_file(f)
Logger.set_project_name(pipeline_configuration.pipeline_name)
log.debug(f"Pipeline name is {pipeline_configuration.pipeline_name}")
uploaded_memory_logs = google_cloud_utils.list_blobs(google_cloud_credentials_file_path,
pipeline_configuration.memory_profile_upload_bucket,
pipeline_configuration.bucket_dir_path, )
uploaded_memory_log_dates = get_uploaded_file_dates(uploaded_memory_logs, date_pattern)
uploaded_data_archives = google_cloud_utils.list_blobs(google_cloud_credentials_file_path,
pipeline_configuration.data_archive_upload_bucket,
pipeline_configuration.bucket_dir_path)
uploaded_data_archives_dates = get_uploaded_file_dates(uploaded_data_archives, date_pattern)
log.warning(f"Deleting old memory profile files from local disk...")
delete_old_log_files(memory_profile_dir_path, uploaded_memory_log_dates)
log.warning(f"Deleting old data archives files from local disk...")
delete_old_log_files(data_archive_dir_path, uploaded_data_archives_dates)
log.info(f"Uploading memory profile files...")
memory_log_files_by_date = get_files_by_date(memory_profile_dir_path, uploaded_memory_log_dates)
for file_date in memory_log_files_by_date:
latest_memory_log_file_path = max(memory_log_files_by_date[file_date], key=os.path.getmtime)
memory_profile_upload_location = f"{pipeline_configuration.memory_profile_upload_bucket}/" \
f"{pipeline_configuration.bucket_dir_path}/{os.path.basename(latest_memory_log_file_path)}"
log.info(f"Uploading memory profile from {latest_memory_log_file_path} to {memory_profile_upload_location}...")
with open(latest_memory_log_file_path, "rb") as f:
google_cloud_utils.upload_file_to_blob(google_cloud_credentials_file_path, memory_profile_upload_location, f)
log.info(f"Uploading data archive files...")
data_archive_files_by_date = get_files_by_date(data_archive_dir_path, uploaded_data_archives_dates)
for file_date in data_archive_files_by_date:
latest_data_archive_file_path = max(data_archive_files_by_date[file_date], key=os.path.getmtime)
data_archive_upload_location = f"{pipeline_configuration.data_archive_upload_bucket}/" \
f"{pipeline_configuration.bucket_dir_path}/{os.path.basename(latest_data_archive_file_path)}"
log.info(f"Uploading data archive from {latest_data_archive_file_path} to {data_archive_upload_location}...")
with open(latest_data_archive_file_path, "rb") as f:
google_cloud_utils.upload_file_to_blob(google_cloud_credentials_file_path, data_archive_upload_location, f)