o }oi"@s`dZddlZddlZddlZddlmZddlmZdgZddZdd Z e d kr.e dSdS) z/A script to check that copyright headers existsN)datetime)Pathzscripts/get_commonvoice_data.pycCsbg}t|D]\}}|dvrq|dr||qd|vr nqg}|D] }|||q%|S)N) r z #import) enumerate startswithappend)_datalines_to_extractilinecommentsrP/home/ubuntu/.local/lib/python3.10/site-packages/tests/check_copyright_header.pyget_top_commentss  rc Cstjdd}|jdddtd|}ttj}d}d}t |d d d }| d }t |}Wdn1s;wYg}t |jd D]} t| tvrSqJt t| d d d  }|} Wdn1skwYt| } t | |krt| d|| qJd} t| D]\} } ttdtj| rd} d}t||dD]-}|dj|d}|| | vrd}n| d}|dd|d}|| | vrd}nq|s|| t| dn=d| | vr| d7} d| | vstd|D]!}||| | |dvr|| t| d||nq| r"nq| s2t| d|| qJt |dkrOtd|D]}t|q@tddSdS)Nz+Usage for copyright header insertion script) descriptionz--dirzTPath to source files to add copyright header to. Will recurse through subdirectoriesT)helprequiredtypeiztests/py_cprheader.txtrzutf-8)encodingrz*.pyz2 has less header lines than the copyright templateFzCopyright.*NVIDIA.*r) CURRENT_YEAR.z & AFFILIATES.z had an error with the yearopyrightz missed the line: z/ did not match the regex: `Copyright.*NVIDIA.*`z\check_copyright_headers.py found the following files that might not have a copyright header:)argparseArgumentParser add_argumentstr parse_argsintrtodayyearopenreadsplitlenrdirrglob EXCLUSIONS readlinesrprintr r researchcompile IGNORECASErangeformatsysexit)parserargs current_year starting_yearpython_header_pathoriginalpyheaderpyheader_linesproblematic_filesfilenamedatafoundrr year_goodr% year_line year_line_affj_filerrrmain/s               rH__main__) __doc__rr/r5rpathlibrr,rrH__name__rrrrs  H