Артем (Все сообщения пользователя)

local sqlite3 = require("lsqlite3")

if not PrintDbgStr then
    function PrintDbgStr(s) print(s) end
end

Settings = {
    Name = "!!!DB_Levels_v1",
    A1_Ticker = "sec_code [class_code]",
    A1_Enable = 1,
    A1_SaveSettings = 0, -- Добавлен параметр
    A1_target1 = 0.1,    A1_target2 = 0.1,    A1_target3 = 0.1,    A1_target4 = 0.1,    A1_target5 = 0.1,    A1_target6 = 0.1,
    Profile1 = 0.1,    Profile2 = 0.1,
    C1_Level1 = 0.1,    C1_Level2 = 0.1,    C1_Level3 = 0.1,    C1_Level4 = 0.1,    C1_Level5 = 0.1,    C1_Level6 = 0.1,    C1_Level7 = 0.1,    C1_Level8 = 0.1,    C1_Level9 = 0.1,
    C1_Level10_AlertFlag = 0,    C1_Level10 = 0.1,    C1_Level11 = 0.1,    C1_Level12 = 0.1,    C1_Level13 = 0.1,    C1_Level14 = 0.1,    C1_Level15 = 0.1,
    C1_Level16 = 0.1,    C1_Level17 = 0.1,    C1_Level18 = 0.1,    C1_Level19 = 0.1,    C1_Level20 = 0.1,    C1_Level21 = 0.1,    C1_Level22 = 0.1,    C1_Level23 = 0.1,
    C1_Level24 = 0.1,    C1_Level25 = 0.1,    C1_Level26 = 0.1,    C1_Level27 = 0.1,    C1_Level28 = 0.1,    C1_Level29 = 0.1,
    CD1_Level30 = 0.1,    CD1_Level31 = 0.1,    CD1_Level32 = 0.1,    CD1_Level33 = 0.1,    CD1_Level34 = 0.1,    CD1_Level35 = 0.1,    CD1_Level36 = 0.1,    CD1_Level37 = 0.1,    CD1_Level38 = 0.1,    CD1_Level39 = 0.1,
    CD1_Level30_2 = 0.1,C1_Level19_2 = 0.1,    DD1_Volume = 0.1,
    PER = "", DATE = "", TIME = "",
    line = {
        { Name="A1_target1", Color = RGB(255,0, 0), Type = TYPE_LINE, Width = 6  },
        { Name="A1_target2", Color = RGB(255,0, 0), Type = TYPE_LINE, Width = 6 },
        { Name="A1_target3", Color = RGB(255,0, 0), Type = TYPE_LINE, Width = 6 },
        { Name="A1_target4", Color = RGB(0,128, 64), Type = TYPE_LINE, Width = 6 },
        { Name="A1_target5", Color = RGB(0,128, 64), Type = TYPE_LINE, Width = 6 },
        { Name="A1_target6", Color = RGB(0,128, 64), Type = TYPE_LINE, Width = 6 },
        { Name="Profile1", Color = RGB(149, 142, 14), Type = TYPE_BARS, Width = 2 },
        { Name="Profile2", Color = RGB(149, 142, 14), Type = TYPE_BARS, Width = 2 },
        { Name="C1_Level1", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level2", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level3", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level4", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level5", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level6", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level7", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level8", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level9", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level10_AlertFlag", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level10", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level11", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
      { Name="C1_Level12", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
      { Name="C1_Level13", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level14", Color = RGB(255,128, 64), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level15", Color = RGB(128,128, 192), Type = TYPE_DASHDOT, Width = 1 },
        { Name="C1_Level16", Color = RGB(128,128, 192), Type = TYPE_DASHDOT, Width = 1 },
        { Name="C1_Level17", Color = RGB(128,128, 192), Type = TYPE_DASHDOT, Width = 1 },
        { Name="C1_Level18", Color = RGB(128,128, 192), Type = TYPE_DASHDOT, Width = 1 },
        { Name="C1_Level19", Color = RGB(128,128, 192), Type = TYPE_DASHDOT, Width = 1 },
        { Name="C1_Level20", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level21", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level22", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level23", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level24", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level25", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level26", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level27", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level28", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="C1_Level29", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level30", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level31", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level32", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level33", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level34", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level35", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level36", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level37", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level38", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level39", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="CD1_Level30_2", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
      { Name="C1_Level19_2", Color = RGB(255,0, 255), Type = TYPE_LINE, Width = 2 },
        { Name="DD1_Volume", Color = RGB(128,64, 64), Type = TYPE_DASHDOT, Width = 3 }
    }
}

local function get_db_path()
    local script_path = getScriptPath and getScriptPath() or "."
    local sep = package.config:sub(1,1)
    if script_path:sub(-1) ~= sep then script_path = script_path .. sep end
    local db_path = script_path .. "data.sqlite"
    PrintDbgStr("get_db_path: db_path = " .. db_path)
    return db_path
end

local db_fields = {
    "TICKER", "PER", "DATE", "TIME",
    "A1_target1", "A1_target2", "A1_target3", "A1_target4", "A1_target5", "A1_target6",
    "Profile1", "Profile2",
    "C1_Level1", "C1_Level2", "C1_Level3", "C1_Level4", "C1_Level5", "C1_Level6", "C1_Level7", "C1_Level8", "C1_Level9",
    "C1_Level10_AlertFlag", "C1_Level10", "C1_Level11", "C1_Level12", "C1_Level13", "C1_Level14", "C1_Level15",
    "C1_Level16", "C1_Level17", "C1_Level18", "C1_Level19", "C1_Level20", "C1_Level21", "C1_Level22", "C1_Level23",
    "C1_Level24", "C1_Level25", "C1_Level26", "C1_Level27", "C1_Level28", "C1_Level29",
    "CD1_Level30", "CD1_Level31", "CD1_Level32", "CD1_Level33", "CD1_Level34", "CD1_Level35", "CD1_Level36", "CD1_Level37", "CD1_Level38", "CD1_Level39",
    "CD1_Level30_2", "C1_Level19_2","DD1_Volume"
}

function OnChangeSettings()
    PrintDbgStr("OnChangeSettings: start")

    if Settings.A1_SaveSettings ~= 1 then
        PrintDbgStr("OnChangeSettings: A1_SaveSettings ~= 1, exiting")
        return
    end

    local db_path = get_db_path()
    PrintDbgStr("OnChangeSettings: opening database: " .. db_path)
    local db = sqlite3.open(db_path)
    if not db then
        PrintDbgStr("OnChangeSettings: failed to open database: " .. db_path)
        return
    end

    local ticker = Settings.A1_Ticker
    Settings.TICKER = ticker
    PrintDbgStr("OnChangeSettings: current ticker: " .. tostring(ticker))

    -- Проверяем, есть ли запись с таким тикером
    local sql_check = string.format("SEL ECT Id FR OM data WHERE TICKER = '%s' LIMIT 1", ticker)
    local existing_id = nil
    for row in db:nrows(sql_check) do
        existing_id = row.Id
        PrintDbgStr("OnChangeSettings: found existing record with Id = " .. tostring(existing_id))
        break
    end

    if existing_id then
        -- Обновляем существующую запись
        PrintDbgStr("OnChangeSettings: updating record with Id = " .. tostring(existing_id))
        local set_parts = {}
        for _, field in ipairs(db_fields) do
            if field ~= "TICKER" then
                local val = Settings[field]
                local val_str
                if type(val) == "number" then
                    val_str = tostring(val)
                else
                    val_str = "'" .. tostring(val):gsub("'", "''") .. "'"
                end
                table.insert(set_parts, string.format("%s = %s", field, val_str))
            end
        end
        local sql_update = string.format("UPD ATE data SE T %s WHERE Id = %d", table.concat(set_parts, ", "), existing_id)
        PrintDbgStr("OnChangeSettings: executing UPDATE: " .. sql_update)
        local rc = db:exec(sql_update)
        if rc ~= sqlite3.OK then
            PrintDbgStr("OnChangeSettings: upd ate failed: " .. db:errmsg())
        else
            PrintDbgStr("OnChangeSettings: successfully updated record for ticker " .. ticker)
        end
    else
        -- Вставляем новую запись
        PrintDbgStr("OnChangeSettings: record not found, inserting new")
        local columns = {}
        local values = {}
        for _, field in ipairs(db_fields) do
            table.insert(columns, field)
            local val = Settings[field]
            if type(val) == "number" then
                table.insert(values, tostring(val))
            else
                table.insert(values, "'" .. tostring(val):gsub("'", "''") .. "'")
            end
        end
        local sql_insert = string.format("INS ERT INTO data (%s) VALUES (%s)", table.concat(columns, ", "), table.concat(values, ", "))
        PrintDbgStr("OnChangeSettings: executing INSERT: " .. sql_insert)
        local rc = db:exec(sql_insert)
        if rc ~= sqlite3.OK then
            PrintDbgStr("OnChangeSettings: insert failed: " .. db:errmsg())
        else
            PrintDbgStr("OnChangeSettings: successfully inserted new record for ticker " .. ticker)
        end
    end

    PrintDbgStr("OnChangeSettings: finished")
    db:close()
end

-- Остальной код (Init, LoadSettingsFromDB, OnCalculate) без изменений



function LoadSettingsFromDB(a1_ticker)
    PrintDbgStr("LoadSettingsFromDB: start for A1_Ticker = " .. tostring(a1_ticker))
    local db_path = get_db_path()
    local db = sqlite3.open(db_path)
    if not db then
        PrintDbgStr("LoadSettingsFromDB: failed to open database: " .. db_path)
        return
    end
    local sql = string.format("SEL ECT * FR OM data WH ERE TICKER = '%s' LIMIT 1", a1_ticker)
    PrintDbgStr("LoadSettingsFromDB: sql = " .. sql)
    for row in db:nrows(sql) do
        for k, v in pairs(row) do
            if Settings[k] ~= nil and type(Settings[k]) == "number" then
                local num = tonumber(v)
                if num then
                    Settings[k] = num
                    PrintDbgStr("LoadSettingsFromDB: loaded " .. k .. " = " .. tostring(Settings[k]))
                end
            end
        end
        break
    end
    db:close()
    PrintDbgStr("LoadSettingsFromDB: done")
end

local function get_currentticker_db_path()
    local script_path = getScriptPath and getScriptPath() or "."
    local sep = package.config:sub(1,1)
    if script_path:sub(-1) ~= sep then script_path = script_path .. sep end
    return script_path .. "CurrentTicker.sqlite"
end

function GetLastTickerFromCurrentTicker()
    local db_path = get_currentticker_db_path()
    local db = sqlite3.open(db_path)
    if not db then
        PrintDbgStr("GetLastTickerFromCurrentTicker: failed to open database: " .. db_path)
        return nil, nil
    end

    local sql = "SELE CT sec_code, class_code FR OM CurrentTicker ORDER BY id DESC LIMIT 1"
    local sec_code, class_code = nil, nil
    for row in db:nrows(sql) do
        sec_code = row.sec_code
        class_code = row.class_code
        break
    end
    db:close()
    return sec_code, class_code
end


function Init()
    PrintDbgStr("Init: start (CurrentTicker version)")
    local sec_code, class_code = GetLastTickerFromCurrentTicker()
    if sec_code and class_code then
        local template = sec_code .. " [" .. class_code .. "]"
        Settings.A1_Ticker = template
        PrintDbgStr("Init: A1_Ticker se t to: " .. Settings.A1_Ticker)
        LoadSettingsFromDB(Settings.A1_Ticker)
    else
        PrintDbgStr("Init: No ticker found in CurrentTicker, fallback to default")
        Settings.A1_Ticker = "sec_code [class_code]"
    end
    PrintDbgStr("Init: done")
    return #Settings.line
end

function OnCalculate(index)
    local E = Settings.A1_Enable or 1
    return
        E * Settings.A1_target1, E * Settings.A1_target2, E * Settings.A1_target3, E * Settings.A1_target4, E * Settings.A1_target5, E * Settings.A1_target6,
        E * Settings.Profile1, E * Settings.Profile2,
        E * Settings.C1_Level1, E * Settings.C1_Level2, E * Settings.C1_Level3, E * Settings.C1_Level4, E * Settings.C1_Level5, E * Settings.C1_Level6, E * Settings.C1_Level7, E * Settings.C1_Level8, E * Settings.C1_Level9,
        E * Settings.C1_Level10_AlertFlag, E * Settings.C1_Level10, E * Settings.C1_Level11, E * Settings.C1_Level13, E * Settings.C1_Level19,
        E * Settings.C1_Level15, E * Settings.C1_Level16, E * Settings.C1_Level17, E * Settings.C1_Level18, E * Settings.C1_Level19, E * Settings.C1_Level20,
        E * Settings.C1_Level21, E * Settings.C1_Level22, E * Settings.C1_Level23, E * Settings.C1_Level24, E * Settings.C1_Level25, E * Settings.C1_Level26, E * Settings.C1_Level27, E * Settings.C1_Level28, E * Settings.C1_Level29,
        E * Settings.CD1_Level30, E * Settings.CD1_Level31, E * Settings.CD1_Level32, E * Settings.CD1_Level33, E * Settings.CD1_Level34, E * Settings.CD1_Level35, E * Settings.CD1_Level36, E * Settings.CD1_Level37, E * Settings.CD1_Level38, E * Settings.CD1_Level39,
        E * Settings.CD1_Level30_2, E * Settings.C1_Level19, E * Settings.DD1_Volume
end

local sqlite3 = require("lsqlite3")  -- Подключение SQLite (если доступно)

Settings = 
{
    Name = "!!!Current_Ticker",
    line = 
    {
        {
            Name = "FixedLine",
            Color = RGB(0, 255, 0),
            Type = TYPE_LINE,
            Width = 2
        }
    }
}

local db = nil
local saved = false  -- Флаг, чтобы выполнить запись только один раз

function Init()
    PrintDbgStr("!!!Current_Ticker: Init called")
    return 1  -- Количество линий
end

function OnCalculate(index)
    if index == 1 and not saved then
        PrintDbgStr("!!!Current_Ticker: OnCalculate first bar")

        local DSI = getDataSourceInfo()
        if DSI == nil then
            PrintDbgStr("!!!Current_Ticker: getDataSourceInfo() returned nil")
            return 10
        end

        local class_code = DSI.class_code
        local sec_code = DSI.sec_code

        PrintDbgStr("!!!Current_Ticker: class_code=" .. tostring(class_code) .. ", sec_code=" .. tostring(sec_code))

        local info = getSecurityInfo(class_code, sec_code)
        if info == nil then
            PrintDbgStr("!!!Current_Ticker: getSecurityInfo() returned nil")
            return 10
        end

        PrintDbgStr("!!!Current_Ticker: short_name=" .. tostring(info.short_name))

        db = sqlite3.open("C:\\QUIK\\LuaIndicators\\CurrentTicker.sqlite")
        if not db then
            PrintDbgStr("!!!Current_Ticker: Failed to open SQLite database")
            return 10
        end

        local create_table_sql = [[
            CRE ATE   TABLE IF NOT EXISTS CurrentTicker (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                class_code TEXT,
                sec_code TEXT,
                short_name TEXT
            );
        ]]
        db:exec(create_table_sql)
        PrintDbgStr("!!!Current_Ticker: Table CurrentTicker ensured")

        local stmt = db:prepare([[
            INS ERT IN TO CurrentTicker (class_code, sec_code, short_name)
            VALUES (?, ?, ?);
        ]])

        if stmt then
            stmt:bind_values(class_code, sec_code, info.short_name or "")
            stmt:step()
            stmt:finalize()
            PrintDbgStr("!!!Current_Ticker: Data inserted in CurrentTicker table")
        else
            PrintDbgStr("!!!Current_Ticker: Failed to prepare SQLite statement")
        end

        db:close()
        PrintDbgStr("!!!Current_Ticker: Database closed")

        saved = true
    end

    return 10
end

function OnDestroy()
    PrintDbgStr("!!!Current_Ticker: OnDestroy called")
    if db then
        db:close()
        PrintDbgStr("!!!Current_Ticker: Database closed on destroy")
    end
end