ODBCRWExcelDlg.cpp
上传用户:fdpbxz
上传日期:2022-07-13
资源大小:33k
文件大小:12k
源码类别:

数据库编程

开发平台:

Visual C++

  1. // ODBCRWExcelDlg.cpp : implementation file
  2. //
  3. #include "stdafx.h"
  4. #include "ODBCRWExcel.h"
  5. #include "ODBCRWExcelDlg.h"
  6. #ifdef _DEBUG
  7. #define new DEBUG_NEW
  8. #undef THIS_FILE
  9. static char THIS_FILE[] = __FILE__;
  10. #endif
  11. /////////////////////////////////////////////////////////////////////////////
  12. // CAboutDlg dialog used for App About
  13. class CAboutDlg : public CDialog
  14. {
  15. public:
  16. CAboutDlg();
  17. // Dialog Data
  18. //{{AFX_DATA(CAboutDlg)
  19. enum { IDD = IDD_ABOUTBOX };
  20. //}}AFX_DATA
  21. // ClassWizard generated virtual function overrides
  22. //{{AFX_VIRTUAL(CAboutDlg)
  23. protected:
  24. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  25. //}}AFX_VIRTUAL
  26. // Implementation
  27. protected:
  28. //{{AFX_MSG(CAboutDlg)
  29. //}}AFX_MSG
  30. DECLARE_MESSAGE_MAP()
  31. };
  32. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  33. {
  34. //{{AFX_DATA_INIT(CAboutDlg)
  35. //}}AFX_DATA_INIT
  36. }
  37. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  38. {
  39. CDialog::DoDataExchange(pDX);
  40. //{{AFX_DATA_MAP(CAboutDlg)
  41. //}}AFX_DATA_MAP
  42. }
  43. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  44. //{{AFX_MSG_MAP(CAboutDlg)
  45. // No message handlers
  46. //}}AFX_MSG_MAP
  47. END_MESSAGE_MAP()
  48. /////////////////////////////////////////////////////////////////////////////
  49. // CODBCRWExcelDlg dialog
  50. CODBCRWExcelDlg::CODBCRWExcelDlg(CWnd* pParent /*=NULL*/)
  51. : CDialog(CODBCRWExcelDlg::IDD, pParent)
  52. {
  53. //{{AFX_DATA_INIT(CODBCRWExcelDlg)
  54. m_csName = _T("");
  55. m_csAge = _T("");
  56. //}}AFX_DATA_INIT
  57. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  58. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  59. m_csSex = _T("");
  60. m_csDatebaseFile = _T("");
  61. }
  62. void CODBCRWExcelDlg::DoDataExchange(CDataExchange* pDX)
  63. {
  64. CDialog::DoDataExchange(pDX);
  65. //{{AFX_DATA_MAP(CODBCRWExcelDlg)
  66. DDX_Control(pDX, IDC_COMBO_SEX, m_comboSex);
  67. DDX_Control(pDX, IDC_INFO_LIST, m_ListCtrlInfo);
  68. DDX_Text(pDX, IDC_EDIT_NAME, m_csName);
  69. DDX_Text(pDX, IDC_EDIT_AGE, m_csAge);
  70. //}}AFX_DATA_MAP
  71. }
  72. BEGIN_MESSAGE_MAP(CODBCRWExcelDlg, CDialog)
  73. //{{AFX_MSG_MAP(CODBCRWExcelDlg)
  74. ON_WM_SYSCOMMAND()
  75. ON_WM_PAINT()
  76. ON_WM_QUERYDRAGICON()
  77. ON_BN_CLICKED(IDC_BUTTON_READ_EXCEL, OnButtonReadExcel)
  78. ON_BN_CLICKED(IDC_BUTTON_WRITE_EXCEL, OnButtonWriteExcel)
  79. ON_BN_CLICKED(IDC_BUTTON_QUIT, OnButtonQuit)
  80. ON_BN_CLICKED(IDC_BUTTON_CLEAR_LIST, OnButtonClearList)
  81. ON_NOTIFY(NM_CLICK, IDC_INFO_LIST, OnClickInfoList)
  82. //}}AFX_MSG_MAP
  83. END_MESSAGE_MAP()
  84. /////////////////////////////////////////////////////////////////////////////
  85. // CODBCRWExcelDlg message handlers
  86. BOOL CODBCRWExcelDlg::OnInitDialog()
  87. {
  88. CDialog::OnInitDialog();
  89. // Add "About..." menu item to system menu.
  90. // IDM_ABOUTBOX must be in the system command range.
  91. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  92. ASSERT(IDM_ABOUTBOX < 0xF000);
  93. CMenu* pSysMenu = GetSystemMenu(FALSE);
  94. if (pSysMenu != NULL)
  95. {
  96. CString strAboutMenu;
  97. strAboutMenu.LoadString(IDS_ABOUTBOX);
  98. if (!strAboutMenu.IsEmpty())
  99. {
  100. pSysMenu->AppendMenu(MF_SEPARATOR);
  101. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  102. }
  103. }
  104. // Set the icon for this dialog.  The framework does this automatically
  105. //  when the application's main window is not a dialog
  106. SetIcon(m_hIcon, TRUE); // Set big icon
  107. SetIcon(m_hIcon, FALSE); // Set small icon
  108. // TODO: Add extra initialization here
  109. //初始化ListCtrl
  110. m_ListCtrlInfo.SetExtendedStyle( LVS_EX_GRIDLINES );
  111. m_ListCtrlInfo.SetExtendedStyle( m_ListCtrlInfo.GetExtendedStyle() | 
  112.                              LVS_EX_FULLROWSELECT );
  113. m_ListCtrlInfo.InsertColumn(0, "姓名", LVCFMT_CENTER, 185);
  114. m_ListCtrlInfo.InsertColumn(1, "性别", LVCFMT_CENTER, 120);
  115. m_ListCtrlInfo.InsertColumn(2, "年龄", LVCFMT_CENTER, 120);
  116. // 初始化下拉列表框
  117. m_comboSex.AddString("男");
  118. m_comboSex.AddString("女");
  119. m_comboSex.SetCurSel(0);
  120. // 若数据库表没有被创建,则将"读取"按钮和"列表"控件置为无效状态
  121. if (!IsCreatedTable())
  122. {
  123. GetDlgItem(IDC_BUTTON_READ_EXCEL)->EnableWindow(FALSE);
  124. GetDlgItem(IDC_INFO_LIST)->EnableWindow(FALSE);
  125. }
  126. return TRUE;  // return TRUE  unless you set the focus to a control
  127. }
  128. void CODBCRWExcelDlg::OnSysCommand(UINT nID, LPARAM lParam)
  129. {
  130. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  131. {
  132. CAboutDlg dlgAbout;
  133. dlgAbout.DoModal();
  134. }
  135. else
  136. {
  137. CDialog::OnSysCommand(nID, lParam);
  138. }
  139. }
  140. // If you add a minimize button to your dialog, you will need the code below
  141. //  to draw the icon.  For MFC applications using the document/view model,
  142. //  this is automatically done for you by the framework.
  143. void CODBCRWExcelDlg::OnPaint() 
  144. {
  145. if (IsIconic())
  146. {
  147. CPaintDC dc(this); // device context for painting
  148. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  149. // Center icon in client rectangle
  150. int cxIcon = GetSystemMetrics(SM_CXICON);
  151. int cyIcon = GetSystemMetrics(SM_CYICON);
  152. CRect rect;
  153. GetClientRect(&rect);
  154. int x = (rect.Width() - cxIcon + 1) / 2;
  155. int y = (rect.Height() - cyIcon + 1) / 2;
  156. // Draw the icon
  157. dc.DrawIcon(x, y, m_hIcon);
  158. }
  159. else
  160. {
  161. CDialog::OnPaint();
  162. }
  163. }
  164. // The system calls this to obtain the cursor to display while the user drags
  165. //  the minimized window.
  166. HCURSOR CODBCRWExcelDlg::OnQueryDragIcon()
  167. {
  168. return (HCURSOR) m_hIcon;
  169. }
  170. void CODBCRWExcelDlg::OnButtonReadExcel() 
  171. {
  172. // TODO: Add your control notification handler code here
  173. CDatabase database;
  174.     CString csSql = _T("");
  175.     CString csName = _T("");
  176. CString csSex = _T("");
  177. CString csAge = _T("");
  178.     CString csDriver = _T("");
  179.     CString csDsn = _T("");
  180.                                     
  181. // 获取主程序所在路径
  182. // 获取数据库文件
  183. m_csDatebaseFile = GetDatabaseFile();
  184.     // 检索是否安装有Excel驱动 "Microsoft Excel Driver (*.xls)" 
  185.     csDriver = GetExcelDriver();
  186.     if (csDriver.IsEmpty())
  187.     {
  188.         // 没有发现Excel驱动
  189.         AfxMessageBox("没有安装Excel驱动!");
  190.         return;
  191.     }
  192.     
  193.     // 创建进行存取的字符串
  194.     csDsn.Format(_T("ODBC;DRIVER={%s};DSN='';DBQ=%s"), csDriver, m_csDatebaseFile);
  195.     TRY
  196.     {
  197.         // 打开数据库
  198.         database.Open(NULL, false, false, csDsn);
  199.         
  200.         CRecordset rs(&database);
  201.         // 设置读取的查询语句.
  202.         csSql = "SELECT Name, Sex, Age "       
  203.                "FROM ClassmateInfo " ;                
  204.                "ORDER BY Name ";
  205.     
  206.         // 执行查询语句
  207.         rs.Open(CRecordset::forwardOnly, csSql, CRecordset::readOnly);
  208. // 清空列表
  209. if (m_ListCtrlInfo.GetItemCount() !=0)
  210. {
  211. m_ListCtrlInfo.DeleteAllItems();
  212. }
  213.         // 获取查询结果
  214.         while (!rs.IsEOF())
  215.         {
  216.             // 读取Excel内部数值
  217.             rs.GetFieldValue("Name", csName);
  218. rs.GetFieldValue("Sex", csSex);
  219.             rs.GetFieldValue("Age", csAge);
  220. csAge = csAge.Mid(0,csAge.ReverseFind('.'));
  221. // 显示记录的内容
  222. m_ListCtrlInfo.InsertItem(0, _T(""));
  223. m_ListCtrlInfo.SetItemText(0, 0, csName);
  224. m_ListCtrlInfo.SetItemText(0, 1, csSex);
  225. m_ListCtrlInfo.SetItemText(0, 2, csAge);
  226.             // 移到下一条记录
  227.             rs.MoveNext();
  228.         }
  229. // 显示读Excel文件的信息
  230. GetDlgItem(IDC_STATIC_OPERATION_INFO)->SetWindowText(_T("Excel文件读取成功!"));
  231.         // 关闭数据库
  232.         database.Close();                             
  233.     }
  234.     CATCH(CDBException, e)
  235.     {
  236.         // 数据库操作产生异常
  237.         AfxMessageBox("数据库错误: " + e->m_strError);
  238.     }
  239.     END_CATCH;
  240. }
  241. void CODBCRWExcelDlg::OnButtonWriteExcel() 
  242. {
  243. // TODO: Add your control notification handler code here
  244. UpdateData();
  245. // 判决姓名是否为空
  246. if (m_csName.IsEmpty())
  247. {
  248. GetDlgItem(IDC_STATIC_OPERATION_INFO)->SetWindowText(_T("姓名不能为空!"));
  249. return;
  250. }
  251. // 判决年龄是否为空
  252. if (m_csAge.IsEmpty())
  253. {
  254. GetDlgItem(IDC_STATIC_OPERATION_INFO)->SetWindowText(_T("年龄不能为空!"));
  255. return;
  256. }
  257. CDatabase database;
  258. CString csDriver = "MICROSOFT EXCEL DRIVER (*.XLS)"; // Excel安装驱动 
  259. CString csSql = _T("");
  260. // 获取数据库文件
  261. m_csDatebaseFile = GetDatabaseFile();
  262. // 创建进行存取的字符串
  263. csSql.Format("DRIVER={%s};DSN='';FIRSTROWHASNAMES=1;READONLY=FALSE;CREATE_DB="%s";DBQ=%s",
  264.   csDriver, m_csDatebaseFile, m_csDatebaseFile);
  265. TRY
  266. {
  267. // 创建数据库 (Excel表格文件)
  268. if ( database.OpenEx(csSql,CDatabase::noOdbcDialog) )
  269. {
  270. // 判断表ClassmateInfo是否已经被创建,若否,则创建。
  271. if (!IsCreatedTable())
  272. {
  273. // 创建表结构(姓名、性别和年龄)
  274. csSql = "CREATE TABLE ClassmateInfo (Name TEXT,Sex TEXT,Age NUMBER)";
  275. database.ExecuteSQL(csSql);
  276. }  
  277. // 写入ini文件
  278. ::WritePrivateProfileString( "CreateTable", "IsCreated", "TRUE", "./CreateTable.ini" ); 
  279. // 获取性别
  280. int nCurSelSex = m_comboSex.GetCurSel();
  281. m_comboSex.GetLBText(nCurSelSex, m_csSex);
  282. // 插入数值
  283. csSql = "INSERT INTO ClassmateInfo (Name,Sex,Age) VALUES ('"+m_csName+"','"+m_csSex+"','"+m_csAge+"')";
  284. database.ExecuteSQL(csSql);
  285. }  
  286. // 关闭数据库
  287. database.Close();
  288. // 在列表控件中显示记录
  289. OnButtonReadExcel();
  290. // 显示写Excel文件的信息
  291. GetDlgItem(IDC_STATIC_OPERATION_INFO)->SetWindowText(_T("Excel文件写入成功!"));
  292. // 清空编辑框
  293. m_csName = _T("");
  294. m_csAge = _T("");
  295. UpdateData();
  296. GetDlgItem(IDC_BUTTON_READ_EXCEL)->EnableWindow(TRUE);
  297. GetDlgItem(IDC_INFO_LIST)->EnableWindow(TRUE);
  298. }
  299. CATCH_ALL(e)
  300. {
  301. AfxMessageBox("异常发生!");
  302. }
  303. END_CATCH_ALL;
  304. }
  305. CString CODBCRWExcelDlg::GetExcelDriver()
  306. {
  307.     char szBuf[2001];
  308.     WORD cbBufMax = 2000;
  309.     WORD cbBufOut;
  310.     char *pszBuf = szBuf;
  311.     CString csDriver =  _T("");
  312.     // 获取已安装驱动的名称
  313.     if (!SQLGetInstalledDrivers(szBuf, cbBufMax, &cbBufOut))
  314. {
  315.         return _T("");
  316. }
  317.     
  318.     // 检索已安装的驱动是否有Excel
  319.     do
  320.     {
  321.         if (strstr(pszBuf, "Excel") != 0)
  322.         {
  323.             // 发现驱动
  324.             csDriver = CString(pszBuf);
  325.             break;
  326.         }
  327.         pszBuf = strchr(pszBuf, '') + 1;
  328.     }
  329.     while (pszBuf[1] != '');
  330.     return csDriver;
  331. }
  332. BOOL CODBCRWExcelDlg::IsCreatedTable()
  333. {
  334. // 读取CreateTable.ini文件
  335. CString csCreateInfo = _T("");
  336. ::GetPrivateProfileString(_T("CreateTable"), _T("IsCreated"), _T(""), 
  337.                       csCreateInfo.GetBuffer(MAX_PATH), 
  338.   MAX_PATH, 
  339.   _T("./CreateTable.ini"));
  340. if (csCreateInfo.Compare(_T("TRUE")) != 0)
  341. {
  342. return FALSE;
  343. }
  344. return TRUE;
  345. }
  346. void CODBCRWExcelDlg::OnButtonQuit() 
  347. {
  348. // TODO: Add your control notification handler code here
  349. CDialog::OnCancel();
  350. }
  351. void CODBCRWExcelDlg::OnButtonClearList() 
  352. {
  353. // TODO: Add your control notification handler code here
  354. // 判断列表是否为空
  355. int nCount = m_ListCtrlInfo.GetItemCount();
  356. if (nCount == 0)
  357. {
  358. GetDlgItem(IDC_STATIC_OPERATION_INFO)->SetWindowText(_T("列表本来就是空的,点击清除没有意义!"));
  359. return;
  360. }
  361. // 清空列表
  362. m_ListCtrlInfo.DeleteAllItems();
  363. // 显示列表已经被清空的信息
  364. GetDlgItem(IDC_STATIC_OPERATION_INFO)->SetWindowText(_T("列表已经被清空!"));
  365. }
  366. void CODBCRWExcelDlg::OnClickInfoList(NMHDR* pNMHDR, LRESULT* pResult) 
  367. {
  368. // TODO: Add your control notification handler code here
  369. // 判断列表是否为空
  370. if (m_ListCtrlInfo.GetItemCount() == 0)
  371. {
  372. GetDlgItem(IDC_STATIC_OPERATION_INFO)->SetWindowText(_T("当前列表中没有数据!"));
  373. return;
  374. }
  375. // 取得第一条记录的位置
  376. POSITION pos = m_ListCtrlInfo.GetFirstSelectedItemPosition();
  377. if ( NULL == pos )
  378. {
  379. // 若没有选择记录,则给出提示
  380. GetDlgItem(IDC_STATIC_OPERATION_INFO)->SetWindowText(_T("你还没有选择一条记录,请选择!"));
  381. return;
  382. }
  383. while (pos)
  384. // 取得被选择记录的索引
  385. int nIndexInFieldList = m_ListCtrlInfo.GetNextSelectedItem(pos);
  386. // 高亮被的选择记录
  387. m_ListCtrlInfo.SetFocus();
  388. // 取得记录中的子项
  389. m_csName = m_ListCtrlInfo.GetItemText(nIndexInFieldList,0);
  390. m_csSex = m_ListCtrlInfo.GetItemText(nIndexInFieldList,1);
  391. m_csAge = m_ListCtrlInfo.GetItemText(nIndexInFieldList,2);
  392. }
  393. m_comboSex.SetWindowText(m_csSex);
  394. UpdateData(FALSE);
  395. *pResult = 0;
  396. }
  397. CString CODBCRWExcelDlg::GetDatabaseFile()
  398. {
  399. CString csPath = _T("");
  400. // 获取主程序所在路径
  401. GetModuleFileName(NULL,csPath.GetBufferSetLength (MAX_PATH+1),MAX_PATH);
  402. csPath.ReleaseBuffer ();
  403. int nPos = 0;
  404. nPos = csPath.ReverseFind ('\');
  405. csPath = csPath.Left (nPos);
  406. CString csDatebaseFile = csPath + "\PersonalInfo.xls"; // 要创建的Excel文件
  407. return csDatebaseFile;
  408. }