From 99a3d14fcc24f103a8d8c94d9b67cc64dac14812 Mon Sep 17 00:00:00 2001
From: f9n <f0n@protonmail.com>
Date: Wed, 8 Jan 2020 15:22:12 +0300
Subject: [PATCH] Add retention example

---
 examples/retention.py | 97 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 examples/retention.py

diff --git a/examples/retention.py b/examples/retention.py
new file mode 100644
index 00000000..0c034ebc
--- /dev/null
+++ b/examples/retention.py
@@ -0,0 +1,97 @@
+import logging
+from os import environ
+from datetime import datetime, timedelta
+
+from swiftclient.service import SwiftService, SwiftError
+
+CONTAINER_NAME = environ.get("SWIFT_CONTAINER_NAME", "sen_affetsen_ben_affetmem")
+RETENTION = environ.get("SWIFT_CONTAINER_RETENTION", "1w")
+"""
+Options:
+    1s
+    1m
+    1h
+    1d
+    1w
+"""
+
+logging.basicConfig(level=logging.ERROR)
+logging.getLogger("requests").setLevel(logging.CRITICAL)
+logging.getLogger("swiftclient").setLevel(logging.CRITICAL)
+logger = logging.getLogger(__name__)
+
+
+UNITS = {"s": "seconds", "m": "minutes", "h": "hours", "d": "days", "w": "weeks"}
+
+
+def convert_to_seconds(s):
+    """
+        https://stackoverflow.com/questions/3096860/convert-time-string-expressed-as-numbermhdsw-to-seconds-in-python
+    """
+    count = int(s[:-1])
+    unit = UNITS[s[-1]]
+    td = timedelta(**{unit: count})
+    return td.seconds + 60 * 60 * 24 * td.days
+
+
+def filter_object_names_by_seconds(container, seconds):
+    """
+      https://github.com/openstack/python-swiftclient/blob/master/examples/list.py
+    """
+    object_names = []
+    now = datetime.now()
+    old_day = now - timedelta(seconds=seconds)
+    old_day_as_isoformat = old_day.isoformat()
+
+    with SwiftService() as swift:
+        try:
+            list_parts_gen = swift.list(container=container)
+            for page in list_parts_gen:
+                if not page["success"]:
+                    raise page["error"]
+
+                for item in page["listing"]:
+                    name = item["name"]
+                    last_modified = item["last_modified"]
+                    if old_day_as_isoformat > last_modified:
+                        object_names.append(name)
+
+        except SwiftError as e:
+            logger.error(e.value)
+
+    return object_names
+
+
+def delete_objects(container, objects):
+    """
+      https://github.com/openstack/python-swiftclient/blob/master/examples/delete.py
+    """
+    _opts = {"object_dd_threads": 20}
+    with SwiftService(options=_opts) as swift:
+        del_iter = swift.delete(container=container, objects=objects)
+        for del_res in del_iter:
+            c = del_res.get("container", "")
+            o = del_res.get("object", "")
+            a = del_res.get("attempts")
+            if del_res["success"] and not del_res["action"] == "bulk_delete":
+                rd = del_res.get("response_dict")
+                if rd is not None:
+                    t = dict(rd.get("headers", {}))
+                    if t:
+                        print(
+                            "Successfully deleted {0}/{1} in {2} attempts "
+                            "(transaction id: {3})".format(c, o, a, t)
+                        )
+                    else:
+                        print(
+                            "Successfully deleted {0}/{1} in {2} "
+                            "attempts".format(c, o, a)
+                        )
+
+
+if __name__ == "__main__":
+    retention_seconds = convert_to_seconds(RETENTION)
+    object_names = filter_object_names_by_seconds(
+        container=CONTAINER_NAME, seconds=retention_seconds
+    )
+    delete_objects(container=CONTAINER_NAME, objects=object_names)