diff -r 5365af94a8ec -r 4cab983479db VirtualMailManager/handler.py --- a/VirtualMailManager/handler.py Sun Jan 06 19:10:37 2013 +0000 +++ b/VirtualMailManager/handler.py Wed Jan 09 20:55:27 2013 +0000 @@ -451,27 +451,20 @@ dom.save() self._make_domain_dir(dom) - def domain_quotalimit(self, domainname, bytes_, messages=0, force=None): + def domain_quotalimit(self, domainname, bytes_, messages=0, force=False): """Wrapper around Domain.update_quotalimit().""" if not all(isinstance(i, int) for i in (bytes_, messages)): raise TypeError("'bytes_' and 'messages' have to be " "integers or longs.") - if force is not None and force != 'force': - raise DomainError(_("Invalid argument: '%s'") % force, - INVALID_ARGUMENT) + assert isinstance(force, bool) dom = self._get_domain(domainname) quotalimit = QuotaLimit(self._dbh, bytes=bytes_, messages=messages) - if force is None: - dom.update_quotalimit(quotalimit) - else: - dom.update_quotalimit(quotalimit, force=True) + dom.update_quotalimit(quotalimit, force) - def domain_services(self, domainname, force=None, *services): + def domain_services(self, domainname, force=False, *services): """Wrapper around Domain.update_serviceset().""" + assert isinstance(force, bool) kwargs = dict.fromkeys(SERVICES, False) - if force is not None and force != 'force': - raise DomainError(_("Invalid argument: '%s'") % force, - INVALID_ARGUMENT) for service in set(services): if service not in SERVICES: raise DomainError(_("Unknown service: '%s'") % service, @@ -480,19 +473,14 @@ dom = self._get_domain(domainname) serviceset = ServiceSet(self._dbh, **kwargs) - dom.update_serviceset(serviceset, (True, False)[not force]) + dom.update_serviceset(serviceset, force) - def domain_transport(self, domainname, transport, force=None): + def domain_transport(self, domainname, transport, force=False): """Wrapper around Domain.update_transport()""" - if force is not None and force != 'force': - raise DomainError(_("Invalid argument: '%s'") % force, - INVALID_ARGUMENT) + assert isinstance(force, bool) dom = self._get_domain(domainname) trsp = Transport(self._dbh, transport=transport) - if force is None: - dom.update_transport(trsp) - else: - dom.update_transport(trsp, force=True) + dom.update_transport(trsp, force) def domain_note(self, domainname, note): """Wrapper around Domain.update_note()"""