summaryrefslogtreecommitdiffstats
path: root/build/plugins/yql_python_udf.py
diff options
context:
space:
mode:
authorMikhail Borisov <[email protected]>2022-02-10 16:45:40 +0300
committerDaniil Cherednik <[email protected]>2022-02-10 16:45:40 +0300
commit5d50718e66d9c037dc587a0211110b7d25a66185 (patch)
treee98df59de24d2ef7c77baed9f41e4875a2fef972 /build/plugins/yql_python_udf.py
parenta6a92afe03e02795227d2641b49819b687f088f8 (diff)
Restoring authorship annotation for Mikhail Borisov <[email protected]>. Commit 2 of 2.
Diffstat (limited to 'build/plugins/yql_python_udf.py')
-rw-r--r--build/plugins/yql_python_udf.py56
1 files changed, 28 insertions, 28 deletions
diff --git a/build/plugins/yql_python_udf.py b/build/plugins/yql_python_udf.py
index 1454129724f..c4f949d8a94 100644
--- a/build/plugins/yql_python_udf.py
+++ b/build/plugins/yql_python_udf.py
@@ -1,42 +1,42 @@
-from _common import sort_by_keywords
-
-
-def get_or_default(kv, name, default):
- if name in kv:
- return kv[name][0]
- return default
-
-
-def onregister_yql_python_udf(unit, *args):
+from _common import sort_by_keywords
+
+
+def get_or_default(kv, name, default):
+ if name in kv:
+ return kv[name][0]
+ return default
+
+
+def onregister_yql_python_udf(unit, *args):
flat, kv = sort_by_keywords({'NAME': 1, 'RESOURCE_NAME': 1, 'ADD_LIBRA_MODULES': 1}, args)
- assert len(flat) == 0
- name = get_or_default(kv, 'NAME', 'CustomPython')
- resource_name = get_or_default(kv, 'RESOURCE_NAME', name)
+ assert len(flat) == 0
+ name = get_or_default(kv, 'NAME', 'CustomPython')
+ resource_name = get_or_default(kv, 'RESOURCE_NAME', name)
add_libra_modules = get_or_default(kv, 'ADD_LIBRA_MODULES', 'no') == 'yes'
-
- use_arcadia_python = unit.get('USE_ARCADIA_PYTHON') == 'yes'
+
+ use_arcadia_python = unit.get('USE_ARCADIA_PYTHON') == 'yes'
py3 = unit.get('PYTHON3') == 'yes'
-
+
unit.onyql_abi_version(['2', '9', '0'])
unit.onpeerdir(['yql/udfs/common/python/python_udf'])
unit.onpeerdir(['ydb/library/yql/public/udf'])
-
+
if add_libra_modules:
unit.onpeerdir(['quality/user_sessions/libra_arc/noyql'])
unit.onpeerdir(['yql/udfs/quality/libra/module'])
- if use_arcadia_python:
- flavor = 'Arcadia'
- unit.onpeerdir([
- 'library/python/runtime',
+ if use_arcadia_python:
+ flavor = 'Arcadia'
+ unit.onpeerdir([
+ 'library/python/runtime',
'yql/udfs/common/python/main'
] if not py3 else [
'library/python/runtime_py3',
'yql/udfs/common/python/main_py3'
- ])
- else:
- flavor = 'System'
-
+ ])
+ else:
+ flavor = 'System'
+
output_includes = [
'yql/udfs/common/python/python_udf/python_udf.h',
'ydb/library/yql/public/udf/udf_registrator.h',
@@ -44,12 +44,12 @@ def onregister_yql_python_udf(unit, *args):
if add_libra_modules:
output_includes.append('yql/udfs/quality/libra/module/module.h')
- path = name + '.yql_python_udf.cpp'
+ path = name + '.yql_python_udf.cpp'
libra_flag = '1' if add_libra_modules else '0'
unit.onpython([
- 'build/scripts/gen_yql_python_udf.py',
+ 'build/scripts/gen_yql_python_udf.py',
flavor, name, resource_name, path, libra_flag,
- 'OUT', path,
+ 'OUT', path,
'OUTPUT_INCLUDES',
] + output_includes
)