chore: apply license headers to all the source files

search_service/__init__.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import ast
 import importlib
 import os

search_service/api/__init__.py

@@ -0,0 +1,2 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0

search_service/api/base.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from http import HTTPStatus
 from typing import Any, Dict, Iterable  # noqa: F401
 

search_service/api/dashboard.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import logging
 from http import HTTPStatus
 from typing import Iterable, Any
@@ -64,6 +67,7 @@ class SearchDashboardFilterAPI(BaseFilterAPI):
     """
     Search Filter for Dashboard
     """
+
     def __init__(self) -> None:
         super().__init__(schema=SearchDashboardResultSchema,
                          index=DASHBOARD_INDEX)

search_service/api/document.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import logging
 
 from http import HTTPStatus

search_service/api/healthcheck.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from typing import Tuple
 from flasgger import swag_from
 

search_service/api/table.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from http import HTTPStatus
 from typing import Any, Iterable  # noqa: F401
 
@@ -58,6 +61,7 @@ class SearchTableFilterAPI(BaseFilterAPI):
     """
     Search Filter for table
     """
+
     def __init__(self) -> None:
         super().__init__(schema=SearchTableResultSchema,
                          index=TABLE_INDEX)

search_service/api/user.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from http import HTTPStatus
 from typing import Iterable, Any
 

search_service/config.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import os
 
 ELASTICSEARCH_INDEX_KEY = 'ELASTICSEARCH_INDEX'

search_service/exception.py

@@ -1,3 +1,7 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
+
 class NotFoundException(Exception):
     def __init__(self, message: str) -> None:
         super().__init__(message)

search_service/models/__init__.py

@@ -0,0 +1,2 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0

search_service/models/base.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from abc import ABCMeta, abstractmethod
 from typing import Set
 

search_service/models/dashboard.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from typing import Set, List
 
 import attr
@@ -46,8 +49,8 @@ class Meta:
 
 @attr.s(auto_attribs=True, kw_only=True)
 class SearchDashboardResult:
-        total_results: int = attr.ib()
-        results: List[Dashboard] = attr.ib(factory=list)
+    total_results: int = attr.ib()
+    results: List[Dashboard] = attr.ib(factory=list)
 
 
 class SearchDashboardResultSchema(AttrsSchema):

search_service/models/search_result.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from typing import Any, List
 
 

search_service/models/table.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from typing import List, Optional, Set
 
 import attr
@@ -65,8 +68,8 @@ class Meta:
 
 @attr.s(auto_attribs=True, kw_only=True)
 class SearchTableResult:
-        total_results: int = attr.ib()
-        results: List[Table] = attr.ib(factory=list)
+    total_results: int = attr.ib()
+    results: List[Table] = attr.ib(factory=list)
 
 
 class SearchTableResultSchema(AttrsSchema):

search_service/models/tag.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import attr
 
 from marshmallow_annotations.ext.attrs import AttrsSchema

search_service/models/user.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 
 from typing import Optional, Set, List
 
@@ -47,8 +50,8 @@ class Meta:
 
 @attr.s(auto_attribs=True, kw_only=True)
 class SearchUserResult:
-        total_results: int = attr.ib()
-        results: List[User] = attr.ib(factory=list)
+    total_results: int = attr.ib()
+    results: List[User] = attr.ib(factory=list)
 
 
 class SearchUserResultSchema(AttrsSchema):

search_service/proxy/__init__.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from threading import Lock
 
 from flask import current_app

search_service/proxy/atlas.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import logging
 from re import sub
 from typing import Any, List, Dict, Tuple, Optional

search_service/proxy/base.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from abc import ABCMeta, abstractmethod
 from typing import Any, Dict, List, Union
 

search_service/proxy/elasticsearch.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import logging
 import uuid
 import itertools
@@ -87,7 +90,7 @@ def __init__(self, *,
     def _get_search_result(self, page_index: int,
                            client: Search,
                            model: Any,
-                           search_result_model: Any=SearchResult) -> Any:
+                           search_result_model: Any = SearchResult) -> Any:
         """
         Common helper function to get result.
 
@@ -140,7 +143,7 @@ def _search_helper(self, page_index: int,
                        client: Search,
                        query_name: dict,
                        model: Any,
-                       search_result_model: Any =SearchResult) -> Any:
+                       search_result_model: Any = SearchResult) -> Any:
         """
         Constructs Elasticsearch Query DSL to:
           1. Use function score to customize scoring of search result. It currently uses "total_usage" field to score.

search_service/proxy/statsd_utilities.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import logging
 from threading import Lock
 from typing import Any, Dict, Callable  # noqa: F401

search_service/search_wsgi.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import os
 
 from search_service import create_app

setup.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import os
 
 from setuptools import setup, find_packages

tests/__init__.py

@@ -0,0 +1,2 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0

tests/unit/__init__.py

@@ -0,0 +1,2 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0

tests/unit/api/__init__.py

@@ -0,0 +1,2 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0

tests/unit/api/dashboard/__init__.py

@@ -0,0 +1,2 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0

tests/unit/api/dashboard/fixtures.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from search_service.models.dashboard import Dashboard
 
 

tests/unit/api/dashboard/test_search_dashboard_api.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from http import HTTPStatus
 from unittest import TestCase
 

tests/unit/api/dashboard/test_search_dashboard_filter.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import unittest
 
 from http import HTTPStatus

tests/unit/api/document/__init__.py

@@ -0,0 +1,2 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0

tests/unit/api/document/test_document_table_api.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import unittest
 
 from http import HTTPStatus

tests/unit/api/document/test_document_tables_api.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import unittest
 
 from http import HTTPStatus

tests/unit/api/document/test_document_user_api.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import unittest
 
 from http import HTTPStatus

tests/unit/api/document/test_document_users_api.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import unittest
 
 from http import HTTPStatus

tests/unit/api/table/__init__.py

@@ -0,0 +1,2 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0

tests/unit/api/table/fixtures.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from search_service.models.table import Table
 from search_service.models.tag import Tag
 

tests/unit/api/table/test_search_table_api.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from http import HTTPStatus
 from unittest import TestCase
 

tests/unit/api/table/test_search_table_filter.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import unittest
 
 from http import HTTPStatus

tests/unit/proxy/__init__.py

@@ -0,0 +1,2 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0

tests/unit/proxy/test_atlas.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import unittest
 
 from mock import MagicMock, patch

tests/unit/proxy/test_elasticsearch.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import unittest
 
 from unittest.mock import patch, MagicMock

tests/unit/proxy/test_statsd_utilities.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 from mock import patch, MagicMock
 from flask import current_app
 from statsd import StatsClient

tests/unit/test_app.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import unittest
 
 from flask import current_app
@@ -9,6 +12,7 @@ class AppTest(unittest.TestCase):
     """
     Test the service if it can stand-up
     """
+
     def setUp(self) -> None:
         config_module_class = 'search_service.config.LocalConfig'
         self.app = create_app(config_module_class=config_module_class)

tests/unit/test_swagger.py

@@ -1,3 +1,6 @@
+# Copyright Contributors to the Amundsen project.
+# SPDX-License-Identifier: Apache-2.0
+
 import unittest
 
 from typing import Any, Dict