diff options
| author | iaz1607 <iaz1607@yandex-team.ru> | 2022-05-19 16:22:42 +0300 |
|---|---|---|
| committer | iaz1607 <iaz1607@yandex-team.ru> | 2022-05-19 16:22:42 +0300 |
| commit | 9b18c778aaca9023c365db7f100490faa5aa8ceb (patch) | |
| tree | 1ddd95c55abebfa71bdbbe6a01d4a26d9f97288b /build/scripts/java_pack_to_file.py | |
| parent | 617c2536904a4114758f4dd3fa5c8e356d4543fe (diff) | |
| download | ydb-9b18c778aaca9023c365db7f100490faa5aa8ceb.tar.gz | |
Fix Kotlin coverage
ref:486ae0399a478a28300588f7f4d5b8b99619235a
Diffstat (limited to 'build/scripts/java_pack_to_file.py')
| -rw-r--r-- | build/scripts/java_pack_to_file.py | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/build/scripts/java_pack_to_file.py b/build/scripts/java_pack_to_file.py index 8d2aeb93fd..c8ab7c311b 100644 --- a/build/scripts/java_pack_to_file.py +++ b/build/scripts/java_pack_to_file.py @@ -2,7 +2,8 @@ import os import re import optparse -PACKAGE_REGEX = re.compile(r'^\s*package\s+(.*?);', flags=re.MULTILINE | re.DOTALL) +JAVA_PACKAGE_REGEX = re.compile(r'^\s*package\s+(.*?);', flags=re.MULTILINE | re.DOTALL) +KOTLIN_PACKAGE_REGEX = re.compile(r'^\s*package\s+(.*?)^', flags=re.MULTILINE | re.DOTALL) def parse_args(): @@ -14,9 +15,15 @@ def parse_args(): def get_package_name(filename): with open(filename) as afile: - match = PACKAGE_REGEX.search(afile.read()) - if match: - return match.group(1).replace('\n\t ', '').replace('.', '/') + content = afile.read() + if filename.endswith(".kt"): + match = KOTLIN_PACKAGE_REGEX.search(content) + if match: + return match.group(1).strip().replace('.', '/') + else: + match = JAVA_PACKAGE_REGEX.search(content) + if match: + return match.group(1).replace('\n\t ', '').replace('.', '/') return '' |