Skip to content

Commit e0bf618

Browse files
committed
hopefully fix ptpimg uploader
1 parent a1f550b commit e0bf618

1 file changed

Lines changed: 5 additions & 8 deletions

File tree

pythonbits/ptpimg.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -124,14 +124,11 @@ def upload_urls(self, *urls):
124124
return self._perform(files=files)
125125

126126
def upload(self, *images):
127-
for image in images:
128-
if urlparse(image).scheme in ('http', 'https'):
129-
yield self.upload_urls(image)
130-
elif urlparse(image).scheme in ('file', ''):
131-
yield self.upload_files(image)
132-
else:
133-
raise Exception('Unknown image URI scheme '
134-
'{}'.format(urlparse(image).scheme))
127+
files, urls = _partition(images)
128+
if urls:
129+
yield from self.upload_urls(*urls)
130+
if files:
131+
yield from self.upload_files(*files)
135132

136133

137134
def _partition(files_or_urls):

0 commit comments

Comments
 (0)