Модуль:GetField: различия между версиями
Pok (обсуждение | вклад) Нет описания правки |
Pok (обсуждение | вклад) Нет описания правки |
||
| (не показано 20 промежуточных версий этого же участника) | |||
| Строка 216: | Строка 216: | ||
return nil | return nil | ||
end | |||
local function resolve_entry_path_value(data, id, parsedPath) | |||
if type(data) ~= "table" or not parsedPath or not id or id == "" then | |||
return nil | |||
end | |||
local entry = resolve_entry(data, id) | |||
if type(entry) ~= "table" then | |||
return nil | |||
end | |||
return get_by_parsed_path(entry, parsedPath) | |||
end | end | ||
| Строка 396: | Строка 409: | ||
end | end | ||
return table.concat(parts, "|") | return table.concat(parts, "|") | ||
end | |||
local function append_flattened_part(parts, key, value) | |||
if value == nil then | |||
return | |||
end | |||
if type(value) == "table" then | |||
if next(value) == nil then | |||
return | |||
end | |||
local ok, json = pcall(mw.text.jsonEncode, value) | |||
if ok and json then | |||
parts[#parts + 1] = key .. "=" .. to_nowiki(json) | |||
end | |||
return | |||
end | |||
parts[#parts + 1] = key .. "=" .. tostring(value) | |||
end | |||
local function flatten_selected_parts(entry, keys) | |||
if type(entry) ~= "table" or type(keys) ~= "table" then | |||
return {} | |||
end | |||
local parts = {} | |||
local seen = {} | |||
for i = 1, #keys do | |||
local key = keys[i] | |||
if type(key) == "string" and key ~= "" and not seen[key] then | |||
seen[key] = true | |||
append_flattened_part(parts, key, get_by_path(entry, key)) | |||
end | |||
end | |||
return parts | |||
end | end | ||
| Строка 409: | Строка 461: | ||
end | end | ||
local function build_tpl(id, pagePath, tplPath, data) | local function split_template_spec(tplPath, tplArgs) | ||
tplPath = mw.text.unstripNoWiki(tplPath or "") | |||
tplArgs = mw.text.unstripNoWiki(tplArgs or "") | |||
if tplArgs ~= "" and string.sub(tplArgs, 1, 1) == "|" then | |||
tplArgs = string.sub(tplArgs, 2) | |||
end | |||
if tplArgs == "" then | |||
local pipePos = string.find(tplPath, "|", 1, true) | |||
if pipePos then | |||
tplArgs = mw.text.unstripNoWiki(string.sub(tplPath, pipePos + 1)) | |||
if tplArgs ~= "" and string.sub(tplArgs, 1, 1) == "|" then | |||
tplArgs = string.sub(tplArgs, 2) | |||
end | |||
tplPath = string.sub(tplPath, 1, pipePos - 1) | |||
end | |||
end | |||
return tplPath, tplArgs | |||
end | |||
local function build_tpl(id, pagePath, tplPath, data, tplArgs) | |||
if id == "" or pagePath == "" or tplPath == "" then | if id == "" or pagePath == "" or tplPath == "" then | ||
return "" | return "" | ||
| Строка 422: | Строка 495: | ||
local entry = resolve_entry(data, id) | local entry = resolve_entry(data, id) | ||
local extra = flatten_entry(entry) | local extra = flatten_entry(entry) | ||
local extraTplArgs = tplArgs or "" | |||
local templatePath = resolve_template_path(tplPath) | local templatePath = resolve_template_path(tplPath) | ||
local tplStr = "{{Шаблон:" .. templatePath .. "|id=" .. tostring(id) | local tplStr = "{{Шаблон:" .. templatePath | ||
if extraTplArgs ~= "" then | |||
tplStr = tplStr .. "|" .. extraTplArgs | |||
end | |||
tplStr = tplStr .. "|id=" .. tostring(id) | |||
if extra ~= "" then | if extra ~= "" then | ||
tplStr = tplStr .. "|" .. extra | tplStr = tplStr .. "|" .. extra | ||
| Строка 493: | Строка 571: | ||
local entry = resolve_entry(data, id) or {} | local entry = resolve_entry(data, id) or {} | ||
return flatten_entry(entry) | return flatten_entry(entry) | ||
end | |||
function p.flattenFieldSelective(frame) | |||
local args = frame.args or {} | |||
local id = args[1] or "" | |||
local pagePath = args[2] or "" | |||
local keysJson = mw.text.unstripNoWiki(args[3] or args.keys or "") | |||
if id == "" or pagePath == "" or keysJson == "" then | |||
return "" | |||
end | |||
local okKeys, keys = pcall(mw.text.jsonDecode, keysJson) | |||
if not okKeys or type(keys) ~= "table" or #keys == 0 then | |||
return "" | |||
end | |||
local moduleName = get_module_name(pagePath) | |||
local data = load_cached_data(moduleName) | |||
if not data then | |||
return "" | |||
end | |||
local entry = resolve_entry(data, id) or {} | |||
local parts = flatten_selected_parts(entry, keys) | |||
if #parts == 0 then | |||
return "" | |||
end | |||
return table.concat(parts, "|") | |||
end | end | ||
| Строка 598: | Строка 705: | ||
local keyPath = args[3] or "" | local keyPath = args[3] or "" | ||
local tplPath = mw.text.unstripNoWiki(args[4] or "") | local tplPath = mw.text.unstripNoWiki(args[4] or "") | ||
local tplArgs = args.tplArgs or args.templateArgs or "" | |||
local searchType = (args.searchType or ""):lower() | local searchType = (args.searchType or ""):lower() | ||
| Строка 609: | Строка 717: | ||
keyPath = args[2] or "" | keyPath = args[2] or "" | ||
tplPath = mw.text.unstripNoWiki(args[3] or "") | tplPath = mw.text.unstripNoWiki(args[3] or "") | ||
tplArgs = args[4] or args.tplArgs or args.templateArgs or "" | |||
end | end | ||
tplPath, tplArgs = split_template_spec(tplPath, tplArgs) | |||
if pagePath == "" or keyPath == "" or tplPath == "" then | if pagePath == "" or keyPath == "" or tplPath == "" then | ||
| Строка 638: | Строка 749: | ||
matches = {} | matches = {} | ||
for _, idKey in ipairs(ids) do | for _, idKey in ipairs(ids) do | ||
local v = | local v = resolve_entry_path_value(data, idKey, parsedPath) | ||
if is_nonempty_value(v) then | if is_nonempty_value(v) then | ||
matches[#matches + 1] = idKey | matches[#matches + 1] = idKey | ||
| Строка 677: | Строка 788: | ||
if not seen[idKey] then | if not seen[idKey] then | ||
local specific = idsTable[idKey] | local specific = idsTable[idKey] | ||
local specificValue = type(specific) == "table" and get_by_parsed_path(specific, parsedPath) or nil | local specificValue = type(specific) == "table" and get_by_parsed_path(specific, parsedPath) or | ||
nil | |||
if specificValue == nil then | if specificValue == nil then | ||
matches[#matches + 1] = idKey | matches[#matches + 1] = idKey | ||
| Строка 704: | Строка 816: | ||
local out = {} | local out = {} | ||
for _, idKey in ipairs(matches) do | for _, idKey in ipairs(matches) do | ||
local tpl = build_tpl(idKey, pagePath, tplPath, data) | local tpl = build_tpl(idKey, pagePath, tplPath, data, tplArgs) | ||
if tpl ~= "" then | if tpl ~= "" then | ||
out[#out + 1] = tpl | out[#out + 1] = tpl | ||
| Строка 723: | Строка 835: | ||
local pagePath = args[2] or "" | local pagePath = args[2] or "" | ||
local tplPath = mw.text.unstripNoWiki(args[3] or "") | local tplPath = mw.text.unstripNoWiki(args[3] or "") | ||
local tplArgs = args[4] or args.tplArgs or args.templateArgs or "" | |||
tplPath, tplArgs = split_template_spec(tplPath, tplArgs) | |||
if id == "" or pagePath == "" or tplPath == "" then | if id == "" or pagePath == "" or tplPath == "" then | ||
| Строка 737: | Строка 851: | ||
end | end | ||
local tplStr = build_tpl(id, pagePath, tplPath, data) | local tplStr = build_tpl(id, pagePath, tplPath, data, tplArgs) | ||
return preprocess_or_return(frame, tplStr) | return preprocess_or_return(frame, tplStr) | ||
end | end | ||
| Строка 747: | Строка 861: | ||
local generatorId = args[3] or "" | local generatorId = args[3] or "" | ||
local tplPath = mw.text.unstripNoWiki(args[4] or "") | local tplPath = mw.text.unstripNoWiki(args[4] or "") | ||
local tplArgs = args[5] or args.tplArgs or args.templateArgs or "" | |||
tplPath, tplArgs = split_template_spec(tplPath, tplArgs) | |||
if searchId == "" or generatorId == "" or tplPath == "" then | if searchId == "" or generatorId == "" or tplPath == "" then | ||
| Строка 772: | Строка 888: | ||
local out = {} | local out = {} | ||
for _, id in ipairs(ids) do | for _, id in ipairs(ids) do | ||
local tpl = build_tpl(id, pagePath, tplPath, data) | local tpl = build_tpl(id, pagePath, tplPath, data, tplArgs) | ||
if tpl ~= "" then | if tpl ~= "" then | ||
out[#out + 1] = tpl | out[#out + 1] = tpl | ||
| Строка 901: | Строка 1017: | ||
local pagePath = args[1] or "" | local pagePath = args[1] or "" | ||
local tplPath = args[2] or "" | local tplPath = args[2] or "" | ||
local tplArgs = args[3] or args.tplArgs or args.templateArgs or "" | |||
tplPath, tplArgs = split_template_spec(tplPath, tplArgs) | |||
if pagePath == "" or tplPath == "" then | if pagePath == "" or tplPath == "" then | ||
| Строка 920: | Строка 1038: | ||
for idKey in pairs(idsTable) do | for idKey in pairs(idsTable) do | ||
local tpl = build_tpl(idKey, pagePath, tplPath, data) | local tpl = build_tpl(idKey, pagePath, tplPath, data, tplArgs) | ||
if tpl ~= "" then | if tpl ~= "" then | ||
out[#out + 1] = tpl | out[#out + 1] = tpl | ||
| Строка 930: | Строка 1048: | ||
local result = table.concat(out, " ") | local result = table.concat(out, " ") | ||
return preprocess_or_return(frame, result) | return preprocess_or_return(frame, result) | ||
end | |||
local function encode_nowiki_json(value) | |||
local ok, json = pcall(mw.text.jsonEncode, value) | |||
if ok and json then | |||
return to_nowiki(json) | |||
end | |||
return nil | |||
end | |||
local function collect_sorted_keys(tbl, stringOnly) | |||
local keys = {} | |||
for k in pairs(tbl) do | |||
if not stringOnly or type(k) == "string" then | |||
keys[#keys + 1] = k | |||
end | |||
end | |||
table.sort(keys, function(a, b) | |||
return tostring(a) < tostring(b) | |||
end) | |||
return keys | |||
end | |||
local function choose_id_key(obj) | |||
local keys = {} | |||
for k in pairs(obj) do | |||
if type(k) == "string" then | |||
keys[#keys + 1] = k | |||
end | |||
end | |||
if #keys == 0 then | |||
return nil | |||
end | |||
table.sort(keys, function(a, b) | |||
local av = obj[a] | |||
local bv = obj[b] | |||
local aPrimitive = type(av) ~= "table" | |||
local bPrimitive = type(bv) ~= "table" | |||
if aPrimitive ~= bPrimitive then | |||
return aPrimitive | |||
end | |||
return tostring(a) < tostring(b) | |||
end) | |||
return keys[1] | |||
end | |||
local function is_array_of_primitives(tbl) | |||
if type(tbl) ~= "table" or not is_array(tbl) then | |||
return false | |||
end | |||
for _, v in ipairs(tbl) do | |||
if type(v) == "table" then | |||
return false | |||
end | |||
end | |||
return true | |||
end | |||
local function append_table_fields(parts, value, options, prefix) | |||
if type(value) ~= "table" or next(value) == nil then | |||
return | |||
end | |||
local valueIsPrimitiveArray = is_array_of_primitives(value) | |||
if options.skipPrimitiveRoot and is_array_of_primitives(value) then | |||
return | |||
end | |||
if prefix and options.includeJsonAtPrefix then | |||
local json = encode_nowiki_json(value) | |||
if json then | |||
parts[#parts + 1] = prefix .. "=" .. tostring(json) | |||
end | |||
end | |||
local keys = collect_sorted_keys(value, false) | |||
for _, k in ipairs(keys) do | |||
if not (options.nestedKeyMode == "raw" and type(k) == "number") then | |||
local v = value[k] | |||
local key | |||
if prefix then | |||
key = prefix .. "." .. tostring(k) | |||
else | |||
key = tostring(k) | |||
end | |||
if type(v) == "table" then | |||
if is_array_of_primitives(v) then | |||
local json = encode_nowiki_json(v) | |||
if json then | |||
parts[#parts + 1] = key .. "=" .. tostring(json) | |||
end | |||
elseif options.nestedKeyMode == "raw" then | |||
local json = encode_nowiki_json(v) | |||
if json then | |||
parts[#parts + 1] = key .. "=" .. tostring(json) | |||
end | |||
end | |||
if next(v) ~= nil and not is_array_of_primitives(v) then | |||
local childPrefix = (type(k) == "string") and key or nil | |||
append_table_fields(parts, v, options, childPrefix) | |||
end | |||
else | |||
parts[#parts + 1] = key .. "=" .. tostring(v) | |||
end | |||
end | |||
end | |||
end | end | ||
| Строка 936: | Строка 1173: | ||
local jsonStr = mw.text.unstripNoWiki(args[1] or args.json or "") | local jsonStr = mw.text.unstripNoWiki(args[1] or args.json or "") | ||
local tplPath = mw.text.unstripNoWiki(args[2] or args.template or "") | local tplPath = mw.text.unstripNoWiki(args[2] or args.template or "") | ||
local tplArgs = args[3] or args.tplArgs or args.templateArgs or "" | |||
tplPath, tplArgs = split_template_spec(tplPath, tplArgs) | |||
if jsonStr == "" or tplPath == "" then | if jsonStr == "" or tplPath == "" then | ||
| Строка 947: | Строка 1186: | ||
local calls = {} | local calls = {} | ||
local nestedOptions = { | |||
includeJsonAtPrefix = true, | |||
nestedKeyMode = "prefixed", | |||
skipPrimitiveRoot = false, | |||
} | |||
local rawTypeOptions = { | |||
includeJsonAtPrefix = false, | |||
nestedKeyMode = "raw", | |||
skipPrimitiveRoot = true, | |||
} | |||
local function | local function makeCall(obj) | ||
if type(obj) ~= "table" then | |||
return | |||
end | |||
local idKey = choose_id_key(obj) | |||
if | if not idKey then | ||
return | return | ||
end | end | ||
local parts = { | local parts = { "{{Шаблон:" .. resolve_template_path(tplPath) } | ||
if tplArgs ~= "" then | |||
parts[#parts + 1] = tplArgs | |||
end | |||
parts[#parts + 1] = tostring(idKey) | |||
local keys = collect_sorted_keys(obj, true) | |||
for _, k in ipairs(keys) do | |||
local v = obj[k] | |||
if k == idKey then | |||
if type(k) == "string" and string.sub(k, 1, 6) == "!type:" then | |||
if type(v) == "table" then | |||
append_table_fields(parts, v, rawTypeOptions, nil) | |||
elseif v ~= nil then | |||
parts[#parts + 1] = "value=" .. tostring(v) | |||
end | |||
elseif type(v) == "table" then | |||
if is_array_of_primitives(v) then | |||
local json = encode_nowiki_json(v) | |||
if json then | |||
parts[#parts + 1] = "value=" .. tostring(json) | |||
end | |||
end | |||
if next(v) ~= nil then | |||
append_table_fields(parts, v, nestedOptions, k) | |||
append_table_fields(parts, v, rawTypeOptions, nil) | |||
end | |||
elseif v ~= nil then | |||
parts[#parts + 1] = "value=" .. tostring(v) | |||
end | |||
else | |||
if type(v) == "table" then | |||
if next(v) ~= nil then | |||
append_table_fields(parts, v, nestedOptions, k) | |||
end | |||
elseif v ~= nil then | |||
parts[#parts + 1] = k .. "=" .. tostring(v) | |||
end | |||
end | end | ||
end | end | ||
| Строка 974: | Строка 1258: | ||
if is_array(data) then | if is_array(data) then | ||
for _, item in ipairs(data) do | for _, item in ipairs(data) do | ||
makeCall(item) | |||
end | end | ||
else | else | ||
makeCall(data) | |||
end | end | ||
| Строка 990: | Строка 1268: | ||
end | end | ||
return frame:preprocess(table.concat(calls, " ")) | |||
end | end | ||
| Строка 1121: | Строка 1398: | ||
return frame:preprocess(table.concat(out, "\n")) | return frame:preprocess(table.concat(out, "\n")) | ||
else | else | ||
return frame:preprocess(table.concat(out, "")) | return frame:preprocess(table.concat(out, " ")) | ||
end | end | ||
end | end | ||
return p | return p | ||