punishdlg.cpp

来自「一个C++和SQL的图书管理系统 功能不多 大家可以下来」· C++ 代码 · 共 91 行

// PunishDlg.cpp : implementation file
//

#include "stdafx.h"
#include "library.h"
#include "PunishDlg.h"
#include "PunishSet1.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CPunishDlg dialog


CPunishDlg::CPunishDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CPunishDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CPunishDlg)
	m_PunishPrice = _T("");
	//}}AFX_DATA_INIT
}


void CPunishDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CPunishDlg)
	DDX_Text(pDX, IDC_PUNISH, m_PunishPrice);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CPunishDlg, CDialog)
	//{{AFX_MSG_MAP(CPunishDlg)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CPunishDlg message handlers

void CPunishDlg::OnOK() 
{
	// TODO: Add extra validation here
	UpdateData();
	CPunishSet	*m_punishSet;
	m_punishSet=new CPunishSet(&m_PunishDB);
	CString str;

	str.Format("select * from Punish");
	if( !m_punishSet->Open(AFX_DB_USE_DEFAULT_TYPE,str) )
	{
		MessageBox("打开数据库失败!","数据库错误",MB_OK);
		return;
	}

	if(m_punishSet->GetRecordCount()!=0)
	{
		m_punishSet->Edit();
		m_punishSet->m_Type = "超出罚款";
		m_punishSet->m_Money = m_PunishPrice;
		m_punishSet->Update();
		
	}
	else
	{
		m_punishSet->AddNew();
		m_punishSet->m_Type = "超出罚款";
		m_punishSet->m_Money = m_PunishPrice;
		m_punishSet->Update();
		
	}
	m_punishSet->Close();
/*	m_punishSet->Edit();
	
	if(m_PunishPrice.IsEmpty())
	{
		AfxMessageBox("请输入罚款金额");
		return;
	}
	if(m_punishSet->CanUpdate())
	{
		m_punishSet->Update();
	}
	*/
	CDialog::OnOK();
}