Update header verification script

Author: bgklein

batch-cli-extensions/azext_batch/custom.py

@@ -2,7 +2,6 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
-import json
 import multiprocessing
 from azure.cli.core.util import get_file_json
 from knack.log import get_logger

scripts/license/_common.py

@@ -7,21 +7,33 @@
 
 ROOT_DIR = os.path.abspath(os.path.join(os.path.dirname(os.path.realpath(__file__)), '..', '..'))
 
-PY_LICENSE_HEADER = \
+EXTENSION_SDK_LICENSE_HEADER = \
 """# --------------------------------------------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
 """
 
+PYTHON_SDK_LICENSE_HEADER = \
+"""# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for
+# license information.
+#
+# Code generated by Microsoft (R) AutoRest Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is
+# regenerated.
+# --------------------------------------------------------------------------
+"""
+
 env_folders = [
     os.path.join(ROOT_DIR, 'env'),
     os.path.join(ROOT_DIR, 'package-verify-env'),
     os.path.join(ROOT_DIR, '.azure')
 ]
 
 def contains_header(text):
-    return PY_LICENSE_HEADER in text
+    return (EXTENSION_SDK_LICENSE_HEADER in text or PYTHON_SDK_LICENSE_HEADER in text)
 
 def has_shebang(text):
     return text.startswith('#!')