Add quota-used-bytes and quota-available-bytes (RFC4331) (#277)

* Add quota-used-bytes and quota-available-bytes

* Fix incorrect path to disk_usage

* Update CHANGELOG.md

Co-authored-by: Martin Wendt <mar10@wwwendt.de>

CHANGELOG.md

@@ -1,6 +1,7 @@
 # Changelog
 
-## 4.1.1 / Unreleased
+## 4.2.0 / Unreleased
+- #277 Add quota-used-bytes and quota-available-bytes (RFC4331) (@NewbieOrange)
 
 ## 4.1.0 / 2022-11-11
 - #246 Add dir_browser/htdocs folder setup.cfg (for install with `pip install  .`)

wsgidav/dav_provider.py

@@ -156,6 +156,8 @@ class _DAVResource(ABC):
         get_creation_date()
         get_display_name()
         get_etag()
+        get_used_bytes()
+        get_available_bytes()
         get_last_modified()
         support_ranges()
 
@@ -278,6 +280,22 @@ def get_etag(self):
         """
         return None
 
+    def get_used_bytes(self) -> Optional[int]:
+        """Return used bytes of the DAV collection.
+        See http://www.webdav.org/specs/rfc4331.html#quota-used-bytes
+
+        This method COULD be implemented for collection resources.
+        """
+        return None
+
+    def get_available_bytes(self) -> Optional[int]:
+        """Return available bytes of the DAV collection.
+        See http://www.webdav.org/specs/rfc4331.html#quota-available-bytes
+
+        This method COULD be implemented for collection resources.
+        """
+        return None
+
     def get_last_modified(self):
         """Contains the Last-Modified header returned by a GET method without
         accept headers.
@@ -531,6 +549,10 @@ def get_property_names(self, *, is_allprop):
             propNameList.append("{DAV:}getcontentlength")
         if self.get_content_type() is not None:
             propNameList.append("{DAV:}getcontenttype")
+        if self.get_used_bytes() is not None:
+            propNameList.append("{DAV:}quota-used-bytes")
+        if self.get_available_bytes() is not None:
+            propNameList.append("{DAV:}quota-available-bytes")
         if self.get_last_modified() is not None:
             propNameList.append("{DAV:}getlastmodified")
         if self.get_display_name() is not None:
@@ -706,6 +728,10 @@ def get_property_value(self, name):
                     etree.SubElement(resourcetypeEL, "{DAV:}collection")
                     return resourcetypeEL
                 return ""
+            elif name == "{DAV:}quota-used-bytes":
+                return self.get_used_bytes()
+            elif name == "{DAV:}quota-available-bytes":
+                return self.get_available_bytes()
             elif (
                 name == "{DAV:}getlastmodified" and self.get_last_modified() is not None
             ):

wsgidav/fs_dav_provider.py

@@ -195,6 +195,12 @@ def get_directory_info(self):
     def get_etag(self):
         return None
 
+    def get_used_bytes(self):
+        return shutil.disk_usage(self._file_path).used
+
+    def get_available_bytes(self):
+        return shutil.disk_usage(self._file_path).free
+
     def get_last_modified(self):
         return self.file_stat[stat.ST_MTIME]