Add ReplicationRule.MIN_PRIORITY|MAX_PRIORITY

Author: agoncharov-reef

b2sdk/replication/setting.py

@@ -26,10 +26,12 @@ class ReplicationRule:
     name: str
     file_name_prefix: str = ''
     is_enabled: bool = True
-    priority: int = 1
+    priority: int = 128
     include_existing_files: bool = False
 
     REPLICATION_RULE_REGEX: ClassVar = re.compile(r'^[a-zA-Z0-9_\-]{1,64}$')
+    MIN_PRIORITY: ClassVar[int] = 1
+    MAX_PRIORITY: ClassVar[int] = 2147483647
 
     def __post_init__(self):
         if not self.destination_bucket_id:
@@ -38,6 +40,12 @@ def __post_init__(self):
         if not self.REPLICATION_RULE_REGEX.match(self.name):
             raise ValueError('replication rule name is invalid')
 
+        if not (self.MIN_PRIORITY <= self.priority <= self.MAX_PRIORITY):
+            raise ValueError('priority should be within [%d, %d] interval' % (
+                self.MIN_PRIORITY,
+                self.MAX_PRIORITY,
+            ))
+
     def as_dict(self) -> dict:
         return {
             'destinationBucketId': self.destination_bucket_id,

b2sdk/replication/setup.py

@@ -33,8 +33,8 @@
 class ReplicationSetupHelper(metaclass=B2TraceMeta):
     """ class with various methods that help with repliction management """
     PRIORITY_OFFSET: ClassVar[int] = 5  #: how far to to put the new rule from the existing rules
-    DEFAULT_PRIORITY: ClassVar[int] = 128  #: what priority to set if there are no preexisting rules
-    MAX_PRIORITY: ClassVar[int] = 255  #: maximum allowed priority of a replication rule
+    DEFAULT_PRIORITY: ClassVar[int] = ReplicationRule.priority  #: what priority to set if there are no preexisting rules
+    MAX_PRIORITY: ClassVar[int] = ReplicationRule.MAX_PRIORITY  #: maximum allowed priority of a replication rule
     DEFAULT_SOURCE_CAPABILITIES: ClassVar[Tuple[str, ...]] = (
         'readFiles',
         'readFileLegalHolds',