File tree Expand file tree Collapse file tree 4 files changed +6
-5
lines changed
PyFunceble/checker/availability Expand file tree Collapse file tree 4 files changed +6
-5
lines changed Original file line number Diff line number Diff line change 6060import PyFunceble .facility
6161import PyFunceble .factory
6262import PyFunceble .storage
63+ from PyFunceble .checker .availability .extra_rules import ExtraRulesHandler
6364from PyFunceble .checker .availability .params import AvailabilityCheckerParams
6465from PyFunceble .checker .availability .status import AvailabilityCheckerStatus
6566from PyFunceble .checker .base import CheckerBase
@@ -123,6 +124,7 @@ class AvailabilityCheckerBase(CheckerBase):
123124 domain_syntax_checker : Optional [DomainSyntaxChecker ] = None
124125 ip_syntax_checker : Optional [IPSyntaxChecker ] = None
125126 url_syntax_checker : Optional [URLSyntaxChecker ] = None
127+ extra_rules_handler : Optional [ExtraRulesHandler ] = None
126128
127129 _use_extra_rules : bool = False
128130 _use_whois_lookup : bool = False
@@ -158,6 +160,7 @@ def __init__(
158160 self .domain_syntax_checker = DomainSyntaxChecker ()
159161 self .ip_syntax_checker = IPSyntaxChecker ()
160162 self .url_syntax_checker = URLSyntaxChecker ()
163+ self .extra_rules_handler = ExtraRulesHandler ()
161164 self .db_session = db_session
162165
163166 self .params = AvailabilityCheckerParams ()
Original file line number Diff line number Diff line change 5555import PyFunceble .factory
5656import PyFunceble .storage
5757from PyFunceble .checker .availability .base import AvailabilityCheckerBase
58- from PyFunceble .checker .availability .extra_rules import ExtraRulesHandler
5958from PyFunceble .checker .reputation .domain import DomainReputationChecker
6059
6160
@@ -179,7 +178,7 @@ def query_status(
179178 )
180179
181180 if self .use_extra_rules :
182- ExtraRulesHandler (self .status ).start ()
181+ self . extra_rules_handler . set_status (self .status ).start ()
183182
184183 return self
185184
Original file line number Diff line number Diff line change @@ -79,7 +79,7 @@ class ExtraRulesHandler:
7979
8080 http_codes_dataset : Optional [Box ] = None
8181
82- def __init__ (self , status : Optional [AvailabilityCheckerStatus ]) -> None :
82+ def __init__ (self , status : Optional [AvailabilityCheckerStatus ] = None ) -> None :
8383 self .regex_active2inactive = {
8484 r"\.000webhostapp\.com" : [
8585 (self .__switch_to_down_if , 410 ),
Original file line number Diff line number Diff line change 5555import PyFunceble .factory
5656import PyFunceble .storage
5757from PyFunceble .checker .availability .base import AvailabilityCheckerBase
58- from PyFunceble .checker .availability .extra_rules import ExtraRulesHandler
5958from PyFunceble .checker .reputation .ip import IPReputationChecker
6059
6160
@@ -171,7 +170,7 @@ def query_status(
171170 )
172171
173172 if self .use_extra_rules :
174- ExtraRulesHandler (self .status ).start ()
173+ self . extra_rules_handler . set_status (self .status ).start ()
175174
176175 return self
177176
You can’t perform that action at this time.
0 commit comments