HeidiSQL/source/loaddata.pas

unit loaddata;


// -------------------------------------
// Load Textfile into table
// -------------------------------------


interface

uses
  Windows, Messages, SysUtils, Classes, Graphics, Controls, Forms, Dialogs,
  StdCtrls, ExtCtrls, comctrls, Spin, Buttons, CheckLst, Registry;

type
  Tloaddataform = class(TForm)
    bvlBottom: TBevel;
    btnImport: TButton;
    btnCancel: TButton;
    editFilename: TEdit;
    btnOpenFile: TBitBtn;
    chkLowPriority: TCheckBox;
    lblTable: TLabel;
    comboTable: TComboBox;
    lblFilename: TLabel;
    lblFields: TLabel;
    editFieldTerminator: TEdit;
    editFieldEncloser: TEdit;
    editFieldEscaper: TEdit;
    chkFieldsEnclosedOptionally: TCheckBox;
    lblLines: TLabel;
    editLineTerminator: TEdit;
    lblIgnoreLinesCount: TLabel;
    lblColumns: TLabel;
    chklistColumns: TCheckListBox;
    comboDatabase: TComboBox;
    lblDatabase: TLabel;
    OpenDialogCSVFile: TOpenDialog;
    chkReplace: TCheckBox;
    chkIgnore: TCheckBox;
    lblNote: TLabel;
    btnColUp: TBitBtn;
    btnColDown: TBitBtn;
    editIgnoreLines: TEdit;
    updownIgnoreLines: TUpDown;
    lblFieldTerminater: TLabel;
    lblFieldEncloser: TLabel;
    lblFieldEscaper: TLabel;
    lblLineTerminator: TLabel;
    lblIgnoreLines: TLabel;
    procedure btnCancelClick(Sender: TObject);
    procedure FormShow(Sender: TObject);
    procedure comboDatabaseChange(Sender: TObject);
    procedure comboTableChange(Sender: TObject);
    procedure btnImportClick(Sender: TObject);
    procedure btnOpenFileClick(Sender: TObject);
    procedure chkReplaceClick(Sender: TObject);
    procedure chkIgnoreClick(Sender: TObject);
    procedure btnColUpClick(Sender: TObject);
    procedure btnColDownClick(Sender: TObject);
  private
    { Private declarations }
  public
    { Public declarations }
  end;

  function loaddataWindow(AOwner: TComponent): Boolean;

implementation

uses Main, Childwin, helpers, Db;

{$R *.DFM}


{**
  Create form on demand
  @param TComponent Owner of form (should be calling form)
  @return Boolean Form closed using modalresult mrOK
}
function loaddataWindow(AOwner: TComponent): Boolean;
var
  f : Tloaddataform;
begin
  f := Tloaddataform.Create(AOwner);
  Result := (f.ShowModal=mrOK);
  FreeAndNil(f);
end;


procedure Tloaddataform.btnCancelClick(Sender: TObject);
begin
  close;
end;

procedure Tloaddataform.FormShow(Sender: TObject);
var
  tn : TTreeNode;
  i : Integer;
  reg : TRegistry;
begin
  // read dbs and Tables from treeview
  comboDatabase.Items.Clear;
  for i:=0 to Mainform.ChildWin.DBTree.Items.Count-1 do
  begin
    tn := Mainform.ChildWin.DBTree.Items[i];
    if tn.Level = 1 then
      comboDatabase.Items.Add(tn.Text);
  end;

  with comboDatabase do
  begin
    for i:=0 to Items.Count-1 do
      if Items[i] = Mainform.ChildWin.ActualDatabase then
        ItemIndex := i;
    if ItemIndex = -1 then
      ItemIndex := 0;
  end;

  comboDatabaseChange(self);
  reg := TRegistry.Create;
  if reg.OpenKey(REGPATH, true) then
  begin
    // filename
    editFilename.Text := reg.ReadString('loadfilename');
    // Use options from CSV export
    editFieldTerminator.Text := reg.ReadString('CSVSeparator');
    editFieldEncloser.Text := reg.ReadString('CSVEncloser');
    editLineTerminator.Text := reg.ReadString('CSVTerminator');
    // Other options
    if reg.ValueExists('CSVImportFieldsEnclosedOptionally') then
      chkFieldsEnclosedOptionally.Checked := reg.ReadBool('CSVImportFieldsEnclosedOptionally');
    editFieldEscaper.Text := reg.ReadString('CSVImportFieldEscaper');
    if reg.ValueExists('CSVImportIgnoreLines') then
      updownIgnoreLines.Position := reg.ReadInteger('CSVImportIgnoreLines');
    if reg.ValueExists('CSVImportLowPriority') then
      chkLowPriority.Checked := reg.ReadBool('CSVImportLowPriority');
    if reg.ValueExists('CSVImportReplace') then
      chkReplace.Checked := reg.ReadBool('CSVImportReplace');
    if reg.ValueExists('CSVImportIgnore') then
      chkIgnore.Checked := reg.ReadBool('CSVImportIgnore');
  end;

end;


procedure Tloaddataform.comboDatabaseChange(Sender: TObject);
var
  i : Integer;
begin
  // read tables from db
  comboTable.Items.Clear;
  comboTable.Items := Mainform.ChildWin.GetCol( 'SHOW TABLES FROM ' + MainForm.mask( comboDatabase.Text ) );
  with comboTable do
  begin
    for i:=0 to Items.Count-1 do
      if Items[i] = Mainform.ChildWin.ActualTable then
        ItemIndex := i;
    if ItemIndex = -1 then
      ItemIndex := 0;
  end;

  comboTableChange(self);
end;


procedure Tloaddataform.comboTableChange(Sender: TObject);
var
  i : Integer;
  ds : TDataSet;
begin
  // fill columns:
  chklistColumns.Items.Clear;
  if (comboDatabase.Text <> '') and (comboTable.Text <> '') then begin
    ds := Mainform.ChildWin.GetResults( 'SHOW FIELDS FROM ' + mainform.mask(comboDatabase.Text) + '.' +  mainform.mask(comboTable.Text));
    for i:=1 to ds.RecordCount do
    begin
      chklistColumns.Items.Add(ds.Fields[0].AsString);
      ds.Next;
    end;
  end;

  // select all:
  for i:=0 to chklistColumns.Items.Count-1 do
    chklistColumns.checked[i] := true;
end;


procedure Tloaddataform.btnImportClick(Sender: TObject);
var
  query : string;
  col   : TStringList;
  i     : Integer;
  reg   : TRegistry;
begin

  // Save settings
  reg := TRegistry.Create;
  if reg.OpenKey(REGPATH, true) then
  begin
    // filename
    reg.WriteString( 'loadfilename', editFilename.Text );
    // Use options from CSV export
    reg.WriteString( 'CSVSeparator', editFieldTerminator.Text );
    reg.WriteString( 'CSVEncloser', editFieldEncloser.Text );
    reg.WriteString( 'CSVTerminator', editLineTerminator.Text );
    // Other options
    reg.WriteBool( 'CSVImportFieldsEnclosedOptionally', chkFieldsEnclosedOptionally.Checked );
    reg.WriteString( 'CSVImportFieldEscaper', editFieldEscaper.Text );
    reg.WriteInteger( 'CSVImportIgnoreLines', updownIgnoreLines.Position );
    reg.WriteBool( 'CSVImportLowPriority', chkLowPriority.Checked );
    reg.WriteBool( 'CSVImportReplace', chkReplace.Checked );
    reg.WriteBool( 'CSVImportIgnore', chkIgnore.Checked );
  end;

  query := 'LOAD DATA ';

  if chkLowPriority.Checked then
    query := query + 'LOW_PRIORITY ';

  query := query + 'LOCAL INFILE ' + esc(editFilename.Text) + ' ';
  if chkReplace.Checked then
    query := query + 'REPLACE '
  else if chkIgnore.Checked then
    query := query + 'IGNORE ';
  query := query + 'INTO TABLE ' + Mainform.Mask(comboDatabase.Text) + '.' +  Mainform.Mask(comboTable.Text) + ' ';

  // Fields:
  if (editFieldTerminator.Text <> '') or (editFieldEncloser.Text <> '') or (editFieldEscaper.Text <> '') then
    query := query + 'FIELDS ';
  if editFieldTerminator.Text <> '' then
    query := query + 'TERMINATED BY ' + esc(editFieldTerminator.Text) + ' ';
  if editFieldEncloser.Text <> '' then
  begin
    if chkFieldsEnclosedOptionally.Checked then
      query := query + 'OPTIONALLY ';
    query := query + 'ENCLOSED BY ' + esc(editFieldEncloser.Text) + ' ';
  end;
  if editFieldEscaper.Text <> '' then
    query := query + 'ESCAPED BY ' + esc(editFieldEscaper.Text) + ' ';

  // Lines:
  if editLineTerminator.Text <> '' then
    query := query + 'LINES TERMINATED BY ''' + editLineTerminator.Text + ''' ';
  if updownIgnoreLines.Position > 0 then
    query := query + 'IGNORE ' + inttostr(updownIgnoreLines.Position) + ' LINES ';

  col := TStringList.Create;
  for i:=0 to chklistColumns.Items.Count - 1 do
  begin
    if chklistColumns.checked[i] then
      col.Add(Mainform.Mask( chklistColumns.Items[i] ));
  end;

//  if col.Count < ColumnsCheckListBox.Items.Count then
  query := query + '(' + implodestr(',', col) + ')';

  Mainform.ChildWin.ExecUpdateQuery(query);
  close;
end;

procedure Tloaddataform.btnOpenFileClick(Sender: TObject);
begin
  if OpenDialogCSVFile.Execute then
    editfilename.Text := OpenDialogCSVFile.FileName;
end;

procedure Tloaddataform.chkReplaceClick(Sender: TObject);
begin
  if chkReplace.Checked then
    chkIgnore.checked := false;
end;

procedure Tloaddataform.chkIgnoreClick(Sender: TObject);
begin
  if chkIgnore.Checked then
    chkReplace.checked := false;
end;

procedure Tloaddataform.btnColUpClick(Sender: TObject);
var
  strtemp : String;
  strchecked : boolean;
begin
  // move item up!
  with chklistColumns do
    if ItemIndex > -1 then begin
      if ItemIndex > 0 then begin // not first item...
        strtemp := items[ItemIndex-1];
        strchecked := checked[ItemIndex-1];
        // replace old with new item...
        items[ItemIndex-1] := items[ItemIndex];
        checked[ItemIndex-1] := checked[ItemIndex];
        // and set old item to its origin values...
        items[ItemIndex] := strtemp;
        checked[ItemIndex] := strchecked;

        ItemIndex := ItemIndex-1;
      end;
    end;
end;

procedure Tloaddataform.btnColDownClick(Sender: TObject);
var
  strtemp : String;
  strchecked : boolean;
begin
  // move item down!
  with chklistColumns do
    if ItemIndex > -1 then begin
      if ItemIndex < items.count-1 then begin // not last item...
        strtemp := items[ItemIndex+1];
        strchecked := checked[ItemIndex+1];
        // replace old with new item...
        items[ItemIndex+1] := items[ItemIndex];
        checked[ItemIndex+1] := checked[ItemIndex];
        // and set old item to its origin values...
        items[ItemIndex] := strtemp;
        checked[ItemIndex] := strchecked;

        ItemIndex := ItemIndex+1;
      end;
    end;
end;

end.