def get(self, path):
query = self.sess.query(Collection).filter(Collection.name == path)
try:
result = query.one()
except NoResultFound:
raise CollectionDoesNotExist()
return iRODSCollection(self, result)
def test_set_inherit_acl_depth_test(self):
DEPTH = 3 # But test is valid for any DEPTH > 1
for recursionTruth in (True, False):
deepcoll = None
try:
test_coll_path = self.coll_path + "/test"
deepcoll = helpers.make_deep_collection(self.sess,
test_coll_path,
depth=DEPTH,
objects_per_level=2)
acl1 = iRODSAccess('inherit', deepcoll.path)
self.sess.permissions.set(acl1, recursive=recursionTruth)
test_subcolls = set(
iRODSCollection(self.sess.collections, _)
for _ in self.sess.query(Collection).filter(
Like(Collection.name, deepcoll.path + "/%")))
# assert top level collection affected
test_coll = self.sess.collections.get(test_coll_path)
self.assertTrue(test_coll.inheritance)
#
# assert lower level collections affected only for case when recursive = True
subcoll_truths = [(_.inheritance == recursionTruth)
for _ in test_subcolls]
self.assertEqual(len(subcoll_truths), DEPTH - 1)
self.assertTrue(all(subcoll_truths))
finally:
if deepcoll: deepcoll.remove(force=True, recurse=True)
def get(self, path):
query = self.sess.query(Collection).filter(Collection.name == path)
try:
result = query.one()
except NoResultFound:
raise CollectionDoesNotExist()
return iRODSCollection(self, result)
def get(self, request):
name = request.GET['name']
value = request.GET['value']
query_result = self.irods_session.query(Collection).filter(
CollectionMeta.name == name, CollectionMeta.value == value).all()
results = [iRODSCollection(CollectionManager, row) for row in query_result]
return JsonResponse(map(format_subcoll, results), safe=False)
def push_file(from_path: str, to_prefix: str, session: iRODSSession):
if isfile(to_prefix):
print(f"File {to_prefix} already exists, skipping download")
return
else:
print(f"Uploading file '{from_path}' to '{to_prefix}'")
try:
c_result = session.query(Collection).one()
c = iRODSCollection(session.collections, c_result)
session.data_objects.put(from_path, to_prefix)
except CAT_SQL_ERR:
return
def get(self, path):
path = iRODSCollection.normalize_path(path)
filters = [Collection.name == path]
# if a ticket is supplied for this session, try both without and with DataObject join
repeats = (True,False) if hasattr(self.sess,'ticket__') \
else (False,)
for rep in repeats:
query = self.sess.query(Collection).filter(*filters)
try:
result = query.one()
except NoResultFound:
if rep:
filters += [DataObject.id != 0]
continue
raise CollectionDoesNotExist()
return iRODSCollection(self, result)
def test_upload_directory(remote_base_path):
with TemporaryDirectory() as testdir:
file1_name = 'f1.txt'
file2_name = 'f2.txt'
file1_path = join(testdir, file1_name)
file2_path = join(testdir, file2_name)
coll_name = str(uuid.uuid4())
remote_path = join(remote_base_path, coll_name)
try:
# prep collection
create_collection(remote_path, token)
# create files
with open(file1_path, "w") as file1, open(file2_path,
"w") as file2:
file1.write('Hello, 1!')
file2.write('Hello, 2!')
# create iRODS session and get ticket
with iRODSSession(host='data.cyverse.org',
port=1247,
user='******',
password='',
zone='iplant') as session:
ticket = TerrainTicket.get([
join(remote_path, file1_name),
join(remote_path, file2_name)
], 'write', False)
Ticket(session, ticket).supply()
# upload files
irods_store.push_dir(testdir,
remote_path,
session=session,
include_patterns=['.txt'])
# check uploads
coll = session.query(Collection).one()
collection = iRODSCollection(session.collections, coll)
file_names = [o.name for o in collection.data_objects]
print(file_names)
assert file1_name in file_names
assert file2_name in file_names
finally:
delete_collection(remote_path, token)
def test_create_other_zone_user_227_228(self):
usercolls = []
session = self.sess
A_ZONE_NAME = 'otherZone'
A_ZONE_USER = '******'
try:
zoneB = session.zones.create(A_ZONE_NAME,'remote')
zBuser = session.users.create(A_ZONE_USER,'rodsuser', A_ZONE_NAME, '')
usercolls = [ iRODSCollection(session.collections, result) for result in
session.query(Collection).filter(Collection.owner_name == zBuser.name and
Collection.owner_zone == zBuser.zone) ]
self.assertEqual ([(u[User.name],u[User.zone]) for u in session.query(User).filter(User.zone == A_ZONE_NAME)],
[(A_ZONE_USER,A_ZONE_NAME)])
zBuser.remove()
zoneB.remove()
finally:
for p in usercolls:
try:
session.collections.get( p.path )
except CollectionDoesNotExist:
continue
perm = iRODSAccess( 'own', p.path, session.username, session.zone)
session.permissions.set( perm, admin=True)
p.remove(force=True)