Repository navigation
Expand file tree
/
Copy pathexecute.py
More file actions
98 lines (70 loc) · 2.72 KB
/
Copy pathexecute.py
File metadata and controls
98 lines (70 loc) · 2.72 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
# -*- coding: utf-8 -*-
"""
:authors: python273
:license: Apache License, Version 2.0, see LICENSE file
:copyright: (c) 2019 python273
"""
from .utils import sjson_dumps
from .vk_api import VkApi, VkApiMethod
class VkFunction(object):
""" Обертка над методом execute.
:param code: код функции (VKScript)
:param args: список аргументов (будут конвертированы в JSON)
:param clean_args: список raw аргументов (будут вставлены как строки)
:param return_raw: аргумент raw функции VkApi.method
"""
__slots__ = ('code', '_minified_code', 'args', 'clean_args', 'return_raw')
def __init__(self, code, args=None, clean_args=None, return_raw=False):
self.code = code
self._minified_code = minify(code)
self.args = () if args is None else args
self.clean_args = () if clean_args is None else clean_args
self.return_raw = return_raw
def compile(self, args):
compiled_args = {}
for key, value in args.items():
if key in self.clean_args:
compiled_args[key] = str(value)
else:
compiled_args[key] = sjson_dumps(value)
return self._minified_code % compiled_args
def __call__(self, vk, *args, **kwargs):
"""
:param vk: VkApi или VkApiMethod
:param *args:
:param **kwargs:
"""
if not isinstance(vk, (VkApi, VkApiMethod)):
raise TypeError(
'The first arg should be VkApi or VkApiMethod instance'
)
if isinstance(vk, VkApiMethod):
vk = vk._vk
args = parse_args(self.args, args, kwargs)
return vk.method(
'execute',
{'code': self.compile(args)},
raw=self.return_raw
)
def minify(code):
return ''.join(i.strip() for i in code.splitlines())
def parse_args(function_args, args, kwargs):
parsed_args = {}
for arg_name in kwargs.keys():
if arg_name in function_args:
parsed_args[arg_name] = kwargs[arg_name]
else:
raise VkFunctionException(
f"function got an unexpected keyword argument \'{arg_name}\'"
)
args_count = len(args) + len(kwargs)
func_args_count = len(function_args)
if args_count != func_args_count:
raise VkFunctionException(
f"function takes exactly {func_args_count} argument{'s' if func_args_count > 1 else ''} ({args_count} given)"
)
for arg_name, arg_value in zip(function_args, args):
parsed_args[arg_name] = arg_value
return parsed_args
class VkFunctionException(Exception):
pass