-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathcdnupload.py
816 lines (669 loc) · 30.2 KB
/
cdnupload.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
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
"""Upload static files to a directory or CDN, using content-based hashing.
Run "python cdnupload.py -h" for command line help, or see README.rst for
full documentation.
Released under a permissive MIT license (see LICENSE.txt).
Visit the project's website for more details:
https://github.com/benhoyt/cdnupload
"""
from __future__ import print_function
import argparse
import collections
import errno
import fnmatch
import hashlib
import json
import logging
import mimetypes
import os
import re
import shutil
import sys
try:
from urllib.parse import urlparse
except ImportError:
from urlparse import urlparse
__all__ = ['SourceError', 'DestinationError', 'FileSource', 'Destination',
'FileDestination', 'S3Destination', 'upload', 'delete']
__version__ = '1.0.4'
DEFAULT_HASH_LENGTH = 16
LOG_LEVELS = [
('debug', logging.DEBUG),
('verbose', logging.INFO),
('default', logging.WARNING),
('error', logging.ERROR),
('off', logging.CRITICAL),
]
logger = logging.getLogger('cdnupload')
IS_PY2 = sys.version_info < (3, 0)
if IS_PY2:
input = raw_input
class Error(Exception):
"""Base class that all exceptions raised in this module inherit from."""
class SourceError(Error):
"""Raised when an error occurs accessing the source (usually when building
the key map).
"""
def __init__(self, message, error):
self.message = message
self.error = error
def __str__(self):
return '{}: {}'.format(self.message, self.error)
class DeleteAllKeysError(Error):
"""Raised when delete() would delete all keys in destination."""
class DestinationError(Error):
"""Raised when an error occurs accessing the destination (usually
uploading or deleting). Where relevant, includes the destination key in
question.
"""
def __init__(self, message, error, key=None):
self.message = message
self.error = error
self.key = key
def __str__(self):
return '{}: {}'.format(self.message, self.error)
class FileSource(object):
"""Upload source that recursively returns files from directory tree
starting at given root path. See __init__'s docstring for details.
You can subclass this and pass the instance to upload() or delete() if you
want to customize advanced behaviour like hash_file()'s text handling.
"""
IS_TEXT_BYTES = 8000
def __init__(self, root, dot_names=False, include=None, exclude=None,
ignore_walk_errors=False, follow_symlinks=False,
hash_length=DEFAULT_HASH_LENGTH, hash_chunk_size=64*1024,
hash_class=hashlib.sha1, cache_key_map=True, _os_walk=os.walk):
"""Initialize instance for sourcing files from given root directory.
Include directories and files starting with '.' if "dot_names" is True
(exclude them by default). If "include" is specified, only include
relative paths that match "include" (per fnmatch), or one of the
includes if tuple or list is given. If "exclude" is specified, exclude
any relative paths that match "exclude", or one of the excludes if
tuple or list is given.
If ignore_walk_errors is True, ignore listdir errors when walking the
source tree (except for the root directory, which is always considered
an error). If follow_symlinks is True, follow symbolic links in the
source tree (default is not to follow links).
When building a key mapping, "hash_length" characters of the hex
content hash are included in the filename. The file is read in
"hash_chunk_size" blocks when being hashed. "hash_class" is called
to generate the file hashes (you could use hashlib.md5 or something
else instead).
If cache_key_map is False, don't cache the result of build_key_map().
Default is to cache the result so it doesn't need to be rebuilt if
build_key_map() is called again.
"""
self.root = root
self.dot_names = dot_names
if include and not isinstance(include, (tuple, list)):
include = [include]
self.include = include
if exclude and not isinstance(exclude, (tuple, list)):
exclude = [exclude]
self.exclude = exclude
self.ignore_walk_errors = ignore_walk_errors
self.follow_symlinks = follow_symlinks
self.hash_length = hash_length
self.hash_chunk_size = hash_chunk_size
self.hash_class = hash_class
self.cache_key_map = cache_key_map
self._key_map = None
self.os_walk = _os_walk # for easier testing
def __str__(self):
"""Return a human-readable string describing this source."""
return self.root
def open(self, rel_path):
"""Open file at given relative path."""
path = os.path.join(self.root, rel_path)
return open(path, 'rb')
def hash_file(self, rel_path, is_text=None):
"""Read file at given relative path and return content hash as hex
string (hash is SHA-1 hash of content).
If is_text is None, determine whether file is text like Git does (it's
treated as text if there's no NUL byte in first 8000 bytes).
If file is text, the line endings are normalized to LF by stripping
out any CR characters in the input. This is done to avoid hash
differences between line endings on Windows (CR LF) and Linux/macOS
(LF), especially with "automatic" line ending conversion when using
Git or Subversion.
"""
with self.open(rel_path) as file:
chunk = file.read(self.hash_chunk_size)
if is_text is None:
is_text = chunk.find(b'\x00', 0, self.IS_TEXT_BYTES) == -1
hash_obj = self.hash_class()
while chunk:
if is_text:
chunk = chunk.replace(b'\r', b'')
hash_obj.update(chunk)
chunk = file.read(self.hash_chunk_size)
return hash_obj.hexdigest()
def make_key(self, rel_path, file_hash):
"""Convert relative path and file hash to destination key, for
example, a "rel_path" of 'images/logo.png' would become something like
'images/logo_deadbeef12345678.png'.
The number of characters in the hash part of the destiation key is
specified by the "hash_length" initializer argument.
"""
rel_file, ext = os.path.splitext(rel_path)
key = '{}_{:.{}}{}'.format(rel_file, file_hash, self.hash_length, ext)
return key
def walk_files(self):
"""Generate list of relative paths starting at the source root and
walking the directory tree recursively.
Relative paths in the yielded values are canonicalized to always
use use '/' (forward slash) as a path separator, regardless of running
platform.
"""
if isinstance(self.root, bytes):
# Mainly because os.walk() doesn't handle Unicode chars in walked
# paths on Windows if a bytes path is specified (easy on Python 2.x
# with "str")
walk_root = self.root.decode(sys.getfilesystemencoding())
else:
walk_root = self.root
# Ensure that errors while walking are raised as hard errors, unless
# ignore_walk_errors is True or it's an error listing the root dir
# (on Python 2.x on Windows, error.filename includes the '*.*')
def onerror(error):
if (not self.ignore_walk_errors or error.filename == walk_root or
error.filename == os.path.join(walk_root, '*.*')):
raise error
else:
logger.debug('ignoring error scanning source tree: %s', error)
walker = self.os_walk(walk_root, onerror=onerror,
followlinks=self.follow_symlinks)
for root, dirs, files in walker:
if not self.dot_names:
dirs[:] = [d for d in dirs if not d.startswith('.')]
for file in files:
if not self.dot_names and file.startswith('.'):
continue
rel_path = os.path.relpath(os.path.join(root, file), walk_root)
rel_path = rel_path.replace('\\', '/')
if self.include and not any(fnmatch.fnmatch(rel_path, i)
for i in self.include):
continue
if self.exclude and any(fnmatch.fnmatch(rel_path, e)
for e in self.exclude):
continue
yield rel_path
def build_key_map(self):
"""Walk directory tree starting at source root and build a dict that
maps relative path to key including content-based hash.
The relative paths (keys of the returned dict) are "canonical",
meaning '\' is converted to '/' on Windows, so that users of the
mapping can always look up keys using 'dir/file.ext' style paths,
regardless of operating system.
"""
if self.cache_key_map and self._key_map is not None:
return self._key_map
keys_by_path = {}
for rel_path in self.walk_files():
file_hash = self.hash_file(rel_path)
key = self.make_key(rel_path, file_hash)
keys_by_path[rel_path] = key
if self.cache_key_map:
self._key_map = keys_by_path
return keys_by_path
class Destination(object):
"""Subclass this abstract base class to implement a destination uploader,
for example uploading to Amazon S3, or to Google Cloud Storage.
"""
def __init__(self, destination, **kwargs):
"""Initialize instance with given destination "URL", for example
'/www/static' or 's3://bucket/key/prefix'. Format of destination arg
and names of kwargs depend on the subclass.
"""
raise NotImplementedError
def __str__(self):
"""Return a human-readable string describing this destination, for
example 's3://bucket/key/prefix/'.
"""
raise NotImplementedError
def walk_keys(self):
"""Yield list of keys currently present on the destination"""
raise NotImplementedError
def upload(self, key, source, rel_path):
"""Upload single file from source instance and relative path to
destination at "key".
"""
raise NotImplementedError
def delete(self, key):
"""Delete a single file on the destination at "key"."""
raise NotImplementedError
class FileDestination(Destination):
"""Copies files to a destination directory.
required argument ("destination" command line parameter):
root root of destination directory to copy to
"""
def __init__(self, root):
self.root = root
def __str__(self):
return self.root
def walk_keys(self):
for root, dirs, files in os.walk(self.root):
for file in files:
path = os.path.join(root, file)
key = os.path.relpath(path, self.root)
yield key.replace('\\', '/')
def upload(self, key, source, rel_path):
dest_path = os.path.join(self.root, key)
try:
os.makedirs(os.path.dirname(dest_path))
except OSError as error:
# Because the "exist_ok" param doesn't (ahem) exist on Python 2.x
if error.errno != errno.EEXIST:
raise
with source.open(rel_path) as source_file:
with open(dest_path, 'wb') as dest_file:
shutil.copyfileobj(source_file, dest_file)
def delete(self, key):
os.remove(os.path.join(self.root, key))
class S3Destination(Destination):
"""Uploads files to an Amazon S3 bucket using the boto3 library.
required argument ("destination" command line parameter):
s3_url S3 bucket and key prefix in s3://bucket/key/prefix form
(trailing slash is added to key prefix if not present)
optional arguments:
access_key AWS access key (if not specified, uses credentials in
boto3 environment variables or in ~/.aws/credentials)
secret_key AWS secret key
max_age max-age value for Cache-Control header, in seconds
cache_control full Cache-Control header (overrides max_age)
acl S3 canned ACL (access control list)
region_name AWS region name of S3 bucket (overrides boto3 default or
value in ~/.aws/config)
client_args dict of additional keyword args for boto3 client setup:
boto3.client('s3', ..., **client_args)
upload_args dict of additional keyword args (ExtraArgs) for
client.upload_file() call
"""
def __init__(self, s3_url, access_key=None, secret_key=None,
max_age=365*24*60*60, cache_control='public, max-age={max_age}',
acl='public-read', region_name=None, client_args=None,
upload_args=None, _boto3=None):
parsed = urlparse(s3_url)
if parsed.scheme != 's3':
raise ValueError('s3_url must start with s3://')
if not parsed.netloc:
raise ValueError('s3_url must include a bucket name')
self.bucket_name = parsed.netloc
self.key_prefix = parsed.path.lstrip('/')
if self.key_prefix and not self.key_prefix.endswith('/'):
self.key_prefix += '/'
self.upload_args = upload_args or {}
if acl and 'ACL' not in self.upload_args:
self.upload_args['ACL'] = acl
if cache_control and 'CacheControl' not in self.upload_args:
try:
max_age = int(max_age)
except (ValueError, TypeError):
raise TypeError('max_age must be an integer number of seconds, '
'not {!r}'.format(max_age))
cache_control = cache_control.format(max_age=max_age)
self.upload_args['CacheControl'] = cache_control
# Import boto3 at runtime so it's not required to use cdnupload.py
if _boto3 is None:
try:
import boto3
except ImportError:
raise Exception('boto3 must be installed to upload to S3, try: '
'pip install boto3')
else:
boto3 = _boto3
client_kwargs = dict(
region_name=region_name,
aws_access_key_id=access_key,
aws_secret_access_key=secret_key,
)
if client_args:
client_kwargs.update(client_args)
self.s3_client = boto3.client('s3', **client_kwargs)
def __str__(self):
return 's3://{}/{}'.format(self.bucket_name, self.key_prefix)
def walk_keys(self):
paginator = self.s3_client.get_paginator('list_objects_v2')
pages = paginator.paginate(
Bucket=self.bucket_name,
Prefix=self.key_prefix,
PaginationConfig={'PageSize': 1000},
)
for response in pages:
for obj in response.get('Contents', []):
if obj['Key'].endswith('/'):
# Don't return "folders", empty keys that end with '/'
continue
yield obj['Key']
def upload(self, key, source, rel_path):
content_type = mimetypes.guess_type(rel_path)[0]
key = self.key_prefix + key
extra_args = self.upload_args.copy()
if content_type:
extra_args['ContentType'] = content_type
with source.open(rel_path) as source_file:
self.s3_client.upload_fileobj(source_file, self.bucket_name, key,
ExtraArgs=extra_args)
def delete(self, key):
key = self.key_prefix + key
self.s3_client.delete_object(Bucket=self.bucket_name, Key=key)
# Type returned by top-level upload() and delete() functions
Result = collections.namedtuple('Result', [
'source_key_map',
'destination_keys',
'num_scanned',
'num_processed',
'num_errors',
])
def upload(source, destination, force=False, dry_run=False,
continue_on_errors=False):
"""Upload missing files from source to destination (an instance of a
Destination subclass). Return a Result namedtuple, which includes the
source key map, set of destination keys, and upload statistics.
If "source" is a string, FileSource(source) is used as the source
instance. Otherwise "source" must be a FileSource instance.
The contents of each source file is hashed by the source and included in
the destination key. This is so that if a file changes, it's uploaded
again under a new filename to break caching. For example,
'images/logo.png' will become something like
'images/logo_deadbeef12345678.png'.
If force is True, upload even if files are there already. If dry_run is
True, log what would be uploaded instead of actually uploading.
If continue_on_errors is True, it will continue uploading other files even
if some uploads fail (the default is to raise DestinationError on first
error).
"""
if isinstance(source, (str, bytes)):
source = FileSource(source)
if isinstance(destination, (str, bytes)):
destination = FileDestination(destination)
try:
source_key_map = source.build_key_map()
except Exception as error:
raise SourceError('ERROR scanning source tree', error)
try:
destination_keys = set(destination.walk_keys())
except Exception as error:
raise DestinationError('ERROR listing keys at {}'.format(destination),
error)
options = []
if force:
options.append('force')
if dry_run:
options.append('dry_run')
if continue_on_errors:
options.append('continue_on_errors')
logger.info('starting upload from %s (%d files) to %s (%d existing keys)%s',
source,
len(source_key_map),
destination,
len(destination_keys),
', options: ' + ', '.join(options) if options else '')
num_scanned = 0
num_uploaded = 0
num_errors = 0
for rel_path, key in sorted(source_key_map.items()):
num_scanned += 1
if not force and key in destination_keys:
logger.debug('already uploaded %s, skipping', key)
continue
if key in destination_keys:
verb = 'would force upload' if dry_run else 'force uploading'
else:
verb = 'would upload' if dry_run else 'uploading'
logger.warning('%s %s to %s', verb, rel_path, key)
if not dry_run:
try:
destination.upload(key, source, rel_path)
num_uploaded += 1
except Exception as error:
if not continue_on_errors:
raise DestinationError('ERROR uploading to {}'.format(key),
error, key=key)
logger.error('ERROR uploading to %s: %s', key, error)
num_errors += 1
else:
num_uploaded += 1
logger.info('finished upload: uploaded %d, skipped %d, errors with %d',
num_uploaded, len(source_key_map) - num_uploaded, num_errors)
result = Result(source_key_map, destination_keys,
num_scanned, num_uploaded, num_errors)
return result
def delete(source, destination, force=False, dry_run=False,
continue_on_errors=False):
"""Delete files from destination (an instance of a Destination subclass)
that are no longer present in source tree. Return a Result namedtuple,
which includes the source key map, set of destination keys, and deletion
statistics.
If "source" is a string, FileSource(source) is used as the source
instance. Otherwise "source" must be a FileSource instance.
This function does a sanity check to ensure you're not deleting ALL keys
at the destination by accident (for example, specifying an empty directory
for the source tree). If it would delete all destination keys, it raises
DeleteAllKeysError. To override and delete all anyway, specify force=True.
If dry_run is True, log what would be deleted instead of actually
deleting.
If continue_on_errors is True, it will continue deleting other files even
if some deletes fail (the default is to raise DestinationError on first
error).
"""
if isinstance(source, (str, bytes)):
source = FileSource(source)
if isinstance(destination, (str, bytes)):
destination = FileDestination(destination)
try:
source_key_map = source.build_key_map()
except Exception as error:
raise SourceError('ERROR scanning source tree', error)
source_keys = set(source_key_map.values())
try:
destination_keys = set(destination.walk_keys())
except Exception as error:
raise DestinationError('ERROR listing keys at {}'.format(destination),
error)
options = []
if dry_run:
options.append('dry_run')
if continue_on_errors:
options.append('continue_on_errors')
logger.info('starting delete from %s (%d files) to %s (%d existing keys)%s',
source,
len(source_key_map),
destination,
len(destination_keys),
', options: ' + ', '.join(options) if options else '')
if not force:
num_to_delete = sum(1 for k in destination_keys if k not in source_keys)
if num_to_delete >= len(destination_keys):
raise DeleteAllKeysError(
"ERROR - would delete all {} destination keys, "
"you probably didn't intend this! (use -f/--force or "
"force=True to override)".format(len(destination_keys)))
num_scanned = 0
num_deleted = 0
num_errors = 0
for key in sorted(destination_keys):
num_scanned += 1
if key in source_keys:
logger.debug('still using %s, skipping', key)
continue
verb = 'would delete' if dry_run else 'deleting'
logger.warning('%s %s', verb, key)
if not dry_run:
try:
destination.delete(key)
num_deleted += 1
except Exception as error:
if not continue_on_errors:
raise DestinationError('ERROR deleting {}'.format(key),
error, key=key)
logger.error('ERROR deleting %s: %s', key, error)
num_errors += 1
else:
num_deleted += 1
logger.info('finished delete: deleted %d, errors with %d',
num_deleted, num_errors)
result = Result(source_key_map, destination_keys,
num_scanned, num_deleted, num_errors)
return result
def main(args=None):
"""Command line endpoint for uploading/deleting. If args not specified,
the sys.argv command line arguments are used. Run "cdnupload.py -h" for
detailed help on the arguments.
"""
if args is None:
args = sys.argv[1:]
description = """
cdnupload {version} -- (c) Ben Hoyt 2017 -- github.com/benhoyt/cdnupload
Upload static files from given source directory to destination directory or
Amazon S3 bucket, with content-based hash in filenames for versioning.
""".format(version=__version__)
parser = argparse.ArgumentParser(
description=description,
formatter_class=argparse.RawDescriptionHelpFormatter,
)
parser.add_argument('source',
help='source directory')
parser.add_argument('destination',
help='destination directory (or s3://bucket/path)')
parser.add_argument('dest_args', nargs='*', default=[],
help='optional Destination() keyword args, for example: '
'"max-age=3600"')
parser.add_argument('-a', '--action', default='upload',
choices=['upload', 'delete', 'dest-help'],
help='action to perform (upload, delete, or show help '
'for given Destination class), default %(default)s')
parser.add_argument('-d', '--dry-run', action='store_true',
help='show what script would upload or delete instead of '
'actually doing it')
parser.add_argument('-e', '--exclude', action='append', metavar='PATTERN',
help='exclude source file if its relative path '
'matches, for example *.txt or __pycache__/* '
'(may be specified multiple times)')
parser.add_argument('-f', '--force', action='store_true',
help='force upload even if destination file already exists, '
'or force delete even if it would delete all keys at '
'destination')
parser.add_argument('-i', '--include', action='append', metavar='PATTERN',
help='only include source file if its relative path '
'matches, for example *.png or images/* (may be '
'specified multiple times)')
parser.add_argument('-k', '--key-map', metavar='FILENAME',
help='write source key map to given file as JSON '
'(but only after successful upload or delete)')
parser.add_argument('-l', '--log-level', default='default',
choices=[k for k, v in LOG_LEVELS],
help='set logging level')
parser.add_argument('-v', '--version', action='version', version=__version__)
less_common = parser.add_argument_group('less commonly-used arguments')
less_common.add_argument('--continue-on-errors', action='store_true',
help='continue after upload or delete errors')
less_common.add_argument('--dot-names', action='store_true',
help="include source files and directories starting "
"with '.'")
less_common.add_argument('--follow-symlinks', action='store_true',
help='follow symbolic links when walking source tree')
less_common.add_argument('--hash-length', default=DEFAULT_HASH_LENGTH,
type=int, metavar='N',
help='number of hex chars of hash to use for '
'destination key (default %(default)d)')
less_common.add_argument('--ignore-walk-errors', action='store_true',
help='ignore errors when walking source tree, '
'except for error on root directory')
less_common.add_argument('--license',
help="deprecated (cdnupload now has a simple MIT license)")
args = parser.parse_args()
logging.basicConfig(level=logging.WARNING, format='%(message)s')
log_level = next(v for k, v in LOG_LEVELS if k == args.log_level)
logger.setLevel(log_level)
match = re.match(r'(\w+):', args.destination)
if match:
scheme = match.group(1)
if scheme == 's3':
destination_class = S3Destination
else:
module_name = 'cdnupload_' + scheme
try:
module = __import__(module_name)
except ImportError as error:
parser.error("can't import handler for scheme {!r}: {}".format(
scheme, error))
if not hasattr(module, 'Destination'):
parser.error('{} module has no Destination class'.format(
module_name))
destination_class = getattr(module, 'Destination')
else:
destination_class = FileDestination
if args.action == 'dest-help':
import inspect
import textwrap
arg_spec = inspect.getargspec(destination_class.__init__)
args_str = inspect.formatargspec(*arg_spec)
if args_str.startswith('(self, '):
args_str = '(' + args_str[7:]
args_wrapped = textwrap.fill(
args_str, width=79, initial_indent=' ' * 4,
subsequent_indent=' ' * 5, break_long_words=False,
break_on_hyphens=False)
print('{}'.format(destination_class.__name__))
print(args_wrapped)
print()
print(inspect.getdoc(destination_class))
return 0
source = FileSource(
args.source,
dot_names=args.dot_names,
include=args.include,
exclude=args.exclude,
ignore_walk_errors=args.ignore_walk_errors,
follow_symlinks=args.follow_symlinks,
hash_length=args.hash_length,
)
dest_kwargs = {}
for arg in args.dest_args:
name, sep, value = arg.partition('=')
if not sep:
value = True
name = name.replace('-', '_')
existing = dest_kwargs.get(name)
if existing is not None:
if isinstance(existing, list):
existing.append(value)
else:
dest_kwargs[name] = [existing, value]
else:
dest_kwargs[name] = value
try:
destination = destination_class(args.destination, **dest_kwargs)
except Exception as error:
logger.error('ERROR creating %s instance: %s',
destination_class.__name__, error)
return 1
action_args = dict(
source=source,
destination=destination,
force=args.force,
dry_run=args.dry_run,
continue_on_errors=args.continue_on_errors,
)
try:
if args.action == 'upload':
result = upload(**action_args)
elif args.action == 'delete':
result = delete(**action_args)
else:
assert 'unexpected action {!r}'.format(args.action)
num_errors = result.num_errors
except Error as error:
logger.error('%s', error)
num_errors = 1
if num_errors == 0 and args.key_map:
try:
logger.info('writing key map JSON to {}'.format(args.key_map))
with open(args.key_map, 'w') as f:
json.dump(result.source_key_map, f, sort_keys=True, indent=4)
except Exception as error:
logger.error('ERROR writing key map file: {}'.format(error))
num_errors += 1
return 1 if num_errors else 0
if __name__ == '__main__':
sys.exit(main())