| 27 |
27 |
|
| 28 |
28 |
from optparse import OptionParser
|
| 29 |
29 |
from sys import exit, stdout, stderr
|
|
30 |
from os import environ
|
| 30 |
31 |
|
| 31 |
32 |
try:
|
| 32 |
33 |
from pithos.backends.modular import ModularBackend
|
| ... | ... | |
| 46 |
47 |
|
| 47 |
48 |
def urlsplit(url):
|
| 48 |
49 |
"""Returns (accout, container, object) from a location string"""
|
| 49 |
|
|
|
50 |
|
| 50 |
51 |
assert url.startswith('pithos://'), "Invalid URL"
|
| 51 |
52 |
t = url.split('/', 4)
|
| 52 |
53 |
assert len(t) == 5, "Invalid URL"
|
| ... | ... | |
| 55 |
56 |
|
| 56 |
57 |
def print_size(backend, url):
|
| 57 |
58 |
"""Writes object's size to stdout."""
|
| 58 |
|
|
|
59 |
|
| 59 |
60 |
account, container, object = urlsplit(url)
|
| 60 |
61 |
meta = backend.get_object_meta(account, account, container, object, None)
|
| 61 |
62 |
print meta['bytes']
|
| ... | ... | |
| 63 |
64 |
|
| 64 |
65 |
def print_data(backend, url):
|
| 65 |
66 |
"""Writes object's size to stdout."""
|
| 66 |
|
|
|
67 |
|
| 67 |
68 |
account, container, object = urlsplit(url)
|
| 68 |
69 |
size, hashmap = backend.get_object_hashmap(account, account, container,
|
| 69 |
70 |
object)
|
| ... | ... | |
| 77 |
78 |
|
| 78 |
79 |
def main():
|
| 79 |
80 |
options, args = parser.parse_args()
|
| 80 |
|
if len(args) != 1 or not options.db or not options.data:
|
|
81 |
if len(args) != 1:
|
| 81 |
82 |
parser.print_help()
|
| 82 |
83 |
exit(1)
|
| 83 |
84 |
|
| 84 |
85 |
url = args[0]
|
| 85 |
|
backend = ModularBackend(None, options.db, None, options.data)
|
| 86 |
|
|
|
86 |
|
|
87 |
if not options.data and 'PITHCAT_INPUT_DATA' not in environ:
|
|
88 |
stderr.write("Pithos data directory path is missing.\n")
|
|
89 |
exit(1)
|
|
90 |
|
|
91 |
data_path = environ['PITHCAT_INPUT_DATA'] if not options.data else \
|
|
92 |
options.data
|
|
93 |
|
|
94 |
if not options.db and 'PITHCAT_INPUT_DB' not in environ:
|
|
95 |
stderr.write("Pithos database uri is missing.\n")
|
|
96 |
exit(1)
|
|
97 |
|
|
98 |
db_uri = environ['PITHCAT_INPUT_DB'] if not options.db else options.db
|
|
99 |
|
|
100 |
backend = ModularBackend(None, db_uri, None, data_path)
|
|
101 |
|
| 87 |
102 |
if options.size:
|
| 88 |
103 |
print_size(backend, url)
|
| 89 |
104 |
else:
|