o Ti @sdddlZddlZddlZddlmZdejdZddZddZd d Z e d kr0e dSdS) N)datetimez# Copyright 2020-aN The HuggingFace Team. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. c Cspztjddgtjddd}|j}dd|D}|WStjy7}ztd|gWYd}~Sd}~ww) z1Get a list of all tracked Python files using git.gitzls-filesT)stdouttextcheckcSsg|] }|dr|qS)z.py)endswith).0fr J/home/ubuntu/.local/lib/python3.10/site-packages/scripts/add_copyrights.py -sz,get_tracked_python_files..zError fetching tracked files: N) subprocessrunPIPEr splitlinesCalledProcessErrorprint)resultfilespy_fileser r r get_tracked_python_files%s rcCstj|std|ddSt|dd }|}Wdn1s%wYd|tr4dStd|d t|d dd}| td d|Wdd S1s[wYd S) zECheck if the file contains a copyright notice, and add it if missing.z[SKIP] z does not exist.Nzutf-8)encodingTz[MODIFY] Adding copyright to .w F) ospathisfileropen readlinesjoin startswithCOPYRIGHT_HEADERwrite) file_pathr contentr r r check_and_add_copyright4s   r(cCslt}|s tddStdt|ddd|D}t|s+tdtddStd td dS) zFMain function to check and add copyright for all tracked Python files.z.No Python files are tracked in the repository.Nz Checking z% Python files for copyright notice...cSsg|]}t|qSr )r()rr&r r r r Rszmain..uI❌ Some files were missing the required copyright and have been updated.u*✅ All files have the required copyright.r)rrlenallsysexit)rhave_copyrightr r r mainIsr/__main__) rr r,rnowyearr$rr(r/__name__r r r r s