From 6cee0cd42db555997e6eb8240f1a650828689b3f Mon Sep 17 00:00:00 2001 From: Plamen Dimitrov Date: Wed, 4 May 2022 11:10:24 +0300 Subject: [PATCH] Add case-insensitive replace() behavior to str-inheriting CnfName --- src/cnfvar/model.py | 7 +++++-- 1 files changed, 5 insertions(+), 2 deletions(-) diff --git a/src/cnfvar/model.py b/src/cnfvar/model.py index a8beb5c..768cfa2 100644 --- a/src/cnfvar/model.py +++ b/src/cnfvar/model.py @@ -68,8 +68,11 @@ class CnfName(str): def startswith(self, prefix, *args, **kwargs): return self.lower().startswith(prefix.lower(), *args, **kwargs) - def endswith(self, prefix, *args, **kwargs): - return self.lower().endswith(prefix.lower(), *args, **kwargs) + def endswith(self, suffix, *args, **kwargs): + return self.lower().endswith(suffix.lower(), *args, **kwargs) + + def replace(self, old, new, *args, **kwargs): + return self.lower().replace(old.lower(), new.lower(), *args, **kwargs) ############################################################################### -- 1.7.1