55
66import owlrl
77import rdflib
8- from rdflib import Graph , Literal , URIRef , OWL , RDF , RDFS , XSD
8+ from rdflib import OWL , RDF , RDFS , XSD , Graph , Literal , URIRef
99
1010from ..data_source import set_cli_params as data_source_set_cli_params
11- from ..kgiri import (
12- EKG_NS ,
13- set_kgiri_base ,
14- set_kgiri_base_replace ,
15- set_cli_params as kgiri_set_cli_params ,
16- kgiri_replace_iri_in_literal ,
17- )
18- from ..log import error , warning , log_error , log_list , log_item , log_iri
11+ from ..kgiri import EKG_NS , kgiri_replace_iri_in_literal
12+ from ..kgiri import set_cli_params as kgiri_set_cli_params
13+ from ..kgiri import set_kgiri_base , set_kgiri_base_replace
14+ from ..log import error , log_error , log_iri , log_item , log_list , warning
1915from ..main import load_rdf_file_into_graph
20- from ..namespace import RULE , PROV , RAW , DATAOPS , DATASET
16+ from ..namespace import DATAOPS , DATASET , PROV , RAW , RULE
2117
2218OWL ._fail = (
2319 False # workaround for this issue: https://github.com/RDFLib/OWL-RL/issues/53
@@ -95,11 +91,11 @@ def load_ontology(self, ontology_file_name: Path):
9591 log_item ('Loading Ontology' , ontology_file_name )
9692 load_rdf_file_into_graph (self .g , ontology_file_name )
9793
98- def load_ontologies (self ):
94+ def load_ontologies (self ) -> None :
9995 for ontology_file_name in ontology_file_names :
10096 self .load_ontology (self .ontologies_root / ontology_file_name )
10197
102- def rdfs_infer (self ):
98+ def rdfs_infer (self ) -> None :
10399 owlrl .RDFSClosure .RDFS_Semantics (self .g , True , True , True )
104100 closure_class = owlrl .return_closure_class (
105101 owl_closure = True , rdfs_closure = True , owl_extras = True , trimming = True
@@ -172,10 +168,10 @@ def set_sort_key(self):
172168 return '99-obfuscate'
173169 return f'10-{ set_key } '
174170
175- def key (self ):
171+ def key (self ) -> str :
176172 return f'{ self .set_key ()} -{ self .rule_file .parent .stem } '
177173
178- def sort_key (self ):
174+ def sort_key (self ) -> str :
179175 return f'{ self .set_sort_key ()} -{ self .rule_file .parent .stem } '
180176
181177 def create_rule_set (self ):
@@ -221,7 +217,7 @@ def check_rule(self, rule_iri) -> bool:
221217 # Check to see if in the same directory as the rule.ttl file we have a .sparql_endpoint file
222218 # with the given name. If so, return its contents as a Literal.
223219 #
224- def check_sparql_file_name (self , sparql_file_name ) :
220+ def check_sparql_file_name (self , sparql_file_name : str ) -> Literal | None :
225221 if self .verbose :
226222 log_item ('Checking' , self .rule_file .parent / sparql_file_name )
227223 sparql_file_name_full_path = self .rule_file .parent / sparql_file_name
@@ -233,7 +229,9 @@ def check_sparql_file_name(self, sparql_file_name):
233229 )
234230 return rdflib .Literal (sparql_file_name_full_path .read_text ())
235231
236- def process_sparql_literal (self , rule , sparql_literal ):
232+ def process_sparql_literal (
233+ self , rule : URIRef , sparql_literal : Literal | None
234+ ) -> None :
237235 if not sparql_literal :
238236 return
239237 self .replace_literal_triple (
@@ -249,16 +247,18 @@ def process_sparql_literal(self, rule, sparql_literal):
249247 #
250248 # Add a triple to the graph with the given sparql_endpoint literal.
251249 #
252- def add_literal_triple (self , s , p , o ) :
250+ def add_literal_triple (self , s : URIRef , p : URIRef , o : Literal ) -> None :
253251 if self .verbose :
254252 print ('Adding triple <{0}> - <{1}> - "{2}"' .format (s , p , o ))
255253 self .g .add ((s , p , o ))
256254
257- def replace_literal_triple (self , s , p1 , p2 , o ):
255+ def replace_literal_triple (
256+ self , s : URIRef , p1 : URIRef , p2 : URIRef , o : Literal
257+ ) -> None :
258258 self .g .remove ((s , p1 , None ))
259259 self .add_literal_triple (s , p2 , o )
260260
261- def dump (self , output_file ) -> int :
261+ def dump (self , output_file : str | Path | None ) -> int :
262262 if not output_file :
263263 warning ('You did not specify an output file, no output file created' )
264264 return 1
@@ -311,3 +311,5 @@ def main() -> int:
311311
312312if __name__ == '__main__' :
313313 exit (main ())
314+ if __name__ == '__main__' :
315+ exit (main ())
0 commit comments