Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
s3storage
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package Registry
Container Registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Service Desk
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
cubicweb
cubes
s3storage
Commits
3e07e7b1b10a
Commit
3e07e7b1b10a
authored
2 years ago
by
Efflam Lemaillet
Browse files
Options
Downloads
Patches
Plain Diff
feat: add version id to entity
Refs:
#13
parent
8c41ae0c2e50
No related branches found
No related tags found
No related merge requests found
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
cubicweb_s3storage/storages.py
+35
-9
35 additions, 9 deletions
cubicweb_s3storage/storages.py
with
35 additions
and
9 deletions
cubicweb_s3storage/storages.py
+
35
−
9
View file @
3e07e7b1
...
@@ -86,4 +86,6 @@
...
@@ -86,4 +86,6 @@
self
.
entity_deleted
(
entity
,
attr
)
self
.
entity_deleted
(
entity
,
attr
)
else
:
else
:
oldkey
=
self
.
get_s3_key
(
entity
,
attr
)
oldkey
=
self
.
get_s3_key
(
entity
,
attr
)
if
oldkey
is
not
None
:
oldkey
,
_
=
self
.
parse_key
(
oldkey
)
key
=
self
.
new_s3_key
(
entity
,
attr
)
key
=
self
.
new_s3_key
(
entity
,
attr
)
...
@@ -89,8 +91,4 @@
...
@@ -89,8 +91,4 @@
key
=
self
.
new_s3_key
(
entity
,
attr
)
key
=
self
.
new_s3_key
(
entity
,
attr
)
# save S3 key
entity
.
cw_edited
.
edited_attribute
(
attr
,
Binary
(
key
.
encode
(
'
utf-8
'
)))
# copy Binary value, workaround for boto3 bug
# copy Binary value, workaround for boto3 bug
# https://github.com/boto/s3transfer/issues/80
# https://github.com/boto/s3transfer/issues/80
# .read() is required since Binary can't wrap itself
# .read() is required since Binary can't wrap itself
...
@@ -103,8 +101,8 @@
...
@@ -103,8 +101,8 @@
upload_key
=
key
upload_key
=
key
extra_args
=
self
.
get_upload_extra_args
(
entity
,
attr
,
key
)
extra_args
=
self
.
get_upload_extra_args
(
entity
,
attr
,
key
)
s3
_object
=
self
.
bucket
.
put_object
(
put
_object
_result
=
self
.
s3cnx
.
put_object
(
Body
=
buffer
,
B
ody
=
buffer
,
B
ucket
=
self
.
bucket
,
Key
=
upload_key
,
Key
=
upload_key
,
**
extra_args
)
**
extra_args
)
buffer
.
close
()
buffer
.
close
()
...
@@ -110,4 +108,7 @@
...
@@ -110,4 +108,7 @@
buffer
.
close
()
buffer
.
close
()
version_id
=
put_object_result
.
get
(
'
VersionId
'
,
None
)
# save S3 key
entity
=
self
.
save_s3_key
(
entity
,
attr
,
upload_key
,
version_id
)
# when key is suffixed, move to final key in post commit event
# when key is suffixed, move to final key in post commit event
# remove temporary key on rollback
# remove temporary key on rollback
...
@@ -133,7 +134,7 @@
...
@@ -133,7 +134,7 @@
def
entity_deleted
(
self
,
entity
,
attr
):
def
entity_deleted
(
self
,
entity
,
attr
):
"""
an entity using this storage for attr has been deleted
"""
"""
an entity using this storage for attr has been deleted
"""
key
=
self
.
get_s3_key
(
entity
,
attr
)
key
,
_
=
self
.
parse_key
(
self
.
get_s3_key
(
entity
,
attr
)
)
if
key
is
None
:
if
key
is
None
:
# no key to remove
# no key to remove
return
return
...
@@ -160,6 +161,31 @@
...
@@ -160,6 +161,31 @@
source
.
doexec
(
cnx
,
sql
,
attrs
)
source
.
doexec
(
cnx
,
sql
,
attrs
)
entity
.
cw_edited
=
None
entity
.
cw_edited
=
None
def
save_s3_key
(
self
,
entity
,
attr
,
key
,
version_id
=
None
):
"""
Save the s3 key into the entity bytes attribute
"""
id_string
=
key
if
entity
.
_cw
.
repo
.
config
[
'
s3-activate-object-versioning
'
]
and
\
version_id
is
not
None
:
id_string
=
self
.
format_version_id_suffix
(
key
,
version_id
)
entity
.
cw_edited
.
edited_attribute
(
attr
,
Binary
(
id_string
.
encode
(
'
utf-8
'
)))
return
entity
def
parse_key
(
self
,
key
):
try
:
key
,
version
=
key
.
rsplit
(
self
.
KEY_SEPARATOR
,
1
)
except
ValueError
:
return
key
,
None
return
key
,
version
def
format_version_id_suffix
(
self
,
key
,
version_id
):
"""
Format the string that will store key and version id
"""
return
f
"
{
key
}
#
{
version_id
}
"
def
get_s3_key
(
self
,
entity
,
attr
):
def
get_s3_key
(
self
,
entity
,
attr
):
"""
"""
Return the S3 key of the S3 object storing the content of attribute
Return the S3 key of the S3 object storing the content of attribute
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment