Skip to content

Commit b4a8377

Browse files
committedNov 22, 2019
Restyled by black
1 parent 9f6d469 commit b4a8377

File tree

1 file changed

+20
-8
lines changed

1 file changed

+20
-8
lines changed
 

‎dvc/remote/gdrive/__init__.py

Lines changed: 20 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,11 @@ def gdrive_download_file(
108108
from dvc.progress import Tqdm
109109

110110
gdrive_file = self.drive.CreateFile({"id": file_id})
111-
with Tqdm(desc=progress_name, total=int(gdrive_file["fileSize"]), disable=no_progress_bar):
111+
with Tqdm(
112+
desc=progress_name,
113+
total=int(gdrive_file["fileSize"]),
114+
disable=no_progress_bar,
115+
):
112116
gdrive_file.GetContentFile(to_file)
113117

114118
def gdrive_list_item(self, query):
@@ -132,20 +136,20 @@ def cache_root_dirs(self):
132136

133137
@property
134138
def cached_dirs(self):
135-
if not hasattr(self, '_cached_dirs'):
139+
if not hasattr(self, "_cached_dirs"):
136140
self.drive
137141
return self._cached_dirs
138142

139143
@property
140144
def cached_ids(self):
141-
if not hasattr(self, '_cached_ids'):
145+
if not hasattr(self, "_cached_ids"):
142146
self.drive
143147
return self._cached_ids
144148

145149
@property
146150
@wrap_with(threading.RLock())
147151
def drive(self):
148-
if not hasattr(self, '_gdrive'):
152+
if not hasattr(self, "_gdrive"):
149153
from pydrive.auth import GoogleAuth
150154
from pydrive.drive import GoogleDrive
151155

@@ -214,7 +218,9 @@ def get_remote_item(self, name, parents_ids):
214218
query += " and trashed=false and title='{}'".format(name)
215219

216220
# Limit found remote items count to 1 in response
217-
item_list = self.drive.ListFile({"q": query, "maxResults": 1}).GetList()
221+
item_list = self.drive.ListFile(
222+
{"q": query, "maxResults": 1}
223+
).GetList()
218224
return next(iter(item_list), None)
219225

220226
def resolve_remote_item_from_path(self, parents_ids, path_parts, create):
@@ -240,7 +246,9 @@ def subtract_root_path(self, path_parts):
240246

241247
def get_remote_id_from_cache(self, path_info):
242248
remote_ids = []
243-
path_parts, parents_ids = self.subtract_root_path(path_info.path.split("/"))
249+
path_parts, parents_ids = self.subtract_root_path(
250+
path_info.path.split("/")
251+
)
244252
if (
245253
hasattr(self, "_cached_dirs")
246254
and path_info != self.path_info
@@ -254,12 +262,16 @@ def get_remote_id_from_cache(self, path_info):
254262
return remote_ids, parents_ids, path_parts
255263

256264
def get_remote_id(self, path_info, create=False):
257-
remote_ids, parents_ids, path_parts = self.get_remote_id_from_cache(path_info)
265+
remote_ids, parents_ids, path_parts = self.get_remote_id_from_cache(
266+
path_info
267+
)
258268

259269
if not path_parts and remote_ids:
260270
return remote_ids[0]
261271

262-
file1 = self.resolve_remote_item_from_path(parents_ids, path_parts, create)
272+
file1 = self.resolve_remote_item_from_path(
273+
parents_ids, path_parts, create
274+
)
263275
return file1["id"] if file1 else ""
264276

265277
def exists(self, path_info):

0 commit comments

Comments
 (0)