get_project_size.py

#!/usr/bin/env python
"""
Displays the file size of a DDS Project by ID
Loads config/credentials from ~/.ddsclient
Usage: python3 get_project_size.py c8f46e4d-635a-441a-b309-375818f1f7c1

"""
import sys
from ddsc.config import create_config
from ddsc.core.remotestore import RemoteStore

def main(args=None):
    config = create_config()
    remote_store = RemoteStore(config)
    ds = remote_store.data_service
    project_id = args[1]
    children = ds.get_project_children(project_id, '').json().get('results')
    files = [child for child in children if child['kind'] == 'dds-file']
    total = 0
    for file in files:
      file_size = file['current_version']['upload']['size']
      print('{}\t{}'.format(file_size, file['name']))
      total += file_size
    print()
    print('Project {}: {} bytes ({} gigabytes)'.format(project_id, total, total / (1024 * 1024 * 1024)))

if __name__ == '__main__':
    main(sys.argv)