#!/usr/bin/python

#############################################################
# Created    2003-12-6 
# Modified   2003-12-8
# Project    PostgreSQL Converse
# Model      Create Java Entity Bean (not EJB CMP)
# Company    Organizations
# Author     陈景峰
# Nickname   netkiller
# Email      netkiller@9812.net
# Version    1.1
# Database PostgreSQL 7.3.4
#############################################################

import pg
import string
#db = pg.connect(dbname='netkiller', host='127.0.0.1', user='netkiller',passwd='chen')
db = pg.DB(dbname='netkiller', host='127.0.0.1', user='netkiller',passwd='chen')
# tab = db.get_tables()
# print tab
#print "====================================================="
table = raw_input("Please enter a table: ")
#table = "users"
#classname = "workfile"
classname =  string.capwords(table)

query = "select * from "+table
table_attnames = db.query(query).listfields()
db.close()

#table_attnames = db.get_attnames(table)
buffer = "\n"
buffer += "public class "+classname+"{\n"
buffer += "\n"
for att in table_attnames:
    buffer += "    private String "+ att +" = null;\n"
buffer += "\n"

buffer += "    DBConnect odb = null; \n\n"
buffer += "    private ResultSet rs     = null; \n\n"
buffer += "    private Vector vError = new Vector(); \n\n"
buffer += "    public "+ classname+ "(){\n    }\n"
buffer += "\n"


for att in table_attnames:
    buffer += "    public void set"+ string.capwords(att) +"(String value){\n"
    buffer += "        this."+ att +" = value; \n" 
    buffer += "    }\n"
    buffer += "    public String get"+ string.capwords(att) +"(){\n"
    buffer += "        return this."+ att +"; \n"
    buffer += "    }\n"

buffer += "    public void setPrimaryKey(int id){ \n"
buffer += "        this.id	= id; \n"
buffer += "    	} \n"

finallys = "        finally{ \n"
finallys +="            try{ \n"
finallys +="                if(rs != null)rs.close(); \n"
finallys +="                odb.close(); \n"
finallys +="            }catch(Exception e){ \n"
finallys +="                vError.add(e.toString()); \n"
finallys +="            } \n"
finallys +="        } \n"

buffer += "    public void findByPrimaryKey(){ \n"
buffer += "        String sql = null; \n"
buffer += "        try{ \n"
buffer += "        } \n"
buffer += "        catch(Exception e){ \n"
buffer += "            vError.add(e.toString()); \n"
buffer += "        } \n"
buffer += finallys
buffer += "    	} \n"

buffer += "    public void findAll(){} \n"
buffer += "    public void findByMethod(){} \n"

sql_insert = "insert into " + table + "("
symbol = "?"
prepare = ""
i = 1
for field in table_attnames:
   sql_insert += (field + ",")
   symbol += ",?"
   prepare += "            odb.setString("+ str(i) +",this."+ field +"); \n"
   i += 1
sql_insert += ") values(" + prepare + ")"

buffer += "    public boolean create"+ string.capwords(classname) +"() { \n"
buffer += "        String sql = \""+ sql_insert +"\"; \n"
buffer += "        boolean isSuccess	= false; \n"
buffer += "        try{ \n"
buffer += "            odb = new DBConnect();\n"
buffer += "            odb.prepareStatement(sql); \n"
buffer += prepare
buffer += "            if(odb.executeUpdate()>0) \n"
buffer += "                isSuccess = true; \n"
buffer += "        } \n "
buffer += "        catch(Exception e){ \n"
buffer += "            vError.add(e.toString()); \n"
buffer += "        } \n"
buffer += finallys
buffer += "        return isSuccess; \n"
buffer += "    } \n"

buffer += "    public boolean remove"+ string.capwords(classname) +"() { \n"
buffer += "        String sql = \"delete from "+table+" where id = ? \"; \n";
buffer += "        try{ \n"
buffer += "            odb = new DBConnect(); \n"
buffer += "            odb.prepareStatement(sql); \n"
buffer += "            odb.setInt(1,this.id); \n"
buffer += "            if(odb.executeUpdate()>0) \n"
buffer += "                isSuccess = true; \n"
buffer += "        } \n "
buffer += "        catch(Exception e){ \n"
buffer += "            vError.add(e.toString()); \n"
buffer += "        } \n"
buffer += finallys
buffer += "        return isSuccess; \n"
buffer += "    } \n"

sql_update = "update "+ table +" set "
for field in table_attnames:
   sql_update += (field + "=?,")
sql_update += " where id = ?"

buffer += "    public boolean update"+ string.capwords(classname) +"() { \n"
buffer += "        String sql = \""+ sql_update +"\"; \n"
buffer += "        boolean isSuccess	= false;  \n"
buffer += "        try{ \n"
buffer += "            isSuccess = true; \n"
buffer += "        } \n "
buffer += "        catch(Exception e){ \n"
buffer += "            vError.add(e.toString()); \n"
buffer += "        } \n"
buffer += finallys
buffer += "        return isSuccess; \n"
buffer += "    } \n"

flush = ""
for field in table_attnames:
   flush += ("        this."+field + " = null; \n")

buffer += "    public void flush(){ \n"
buffer += flush
buffer += "    } \n"

buffer += "    public static void main(String[] args) { \n"
buffer += "        " + classname + " test = new " +classname+ "(); \n"
buffer += "    }\n"
buffer +="}\n"


package = "package ebusiness; \n\n"

imports = "import java.sql.*; \n"
imports += "import java.util.*; \n"
imports += "import netkiller.database.*; \n"
imports += "import netkiller.security.*; \n"


f=open(classname+'.java', 'w')
f.write(package);
f.write(imports);
f.write(buffer)
f.close()

#for r in db.query("select * from oa.meeting").dictresult():
#    print '%(id)s %(subject)s' % r
def MakeHtml(html):
    tag = "<!-- db==>html form author:netkiller --> \n"
    tag +="<form name=\"form\" action=\"\" > \n"
    tag += html
    tag += "</form>"
    return tag
def MakeJsp(jsp):
    tag = "<% \n"
    tag += jsp
    tag += "%> \n"
    return tag
html = "<table border=0> \n"
jsp  = "jspParameter para = new jspParameter(request); \n"
jsp += "    String op = request.getParameter(\"op\") \n"
jsp += "    String status = ""; \n"
for att in table_attnames:
    jsp += "    String "+ att +" = para.getString(\""+ att +"\",\"\"); \n"
    html += "    <tr><td>"+ att+":</td><td><input name=\""+ att +"\" type=\"text\" value=\"<%= "+att+" %>\"></td></tr> \n"
jsp += "    if(op != null)){ \n"
jsp += "        if(op.equals(\"create\")){ \n"
jsp += "            status = "<script> alert('添加成功！') </script>" \n"
jsp += "        } \n"
jsp += "        if(op.equals(\"remove\")){} \n"
jsp += "        if(op.equals(\"remove\")){} \n"
jsp += "        if(op.equals(\"update\")){} \n"
jsp += "    } \n"
html += "</table> \n"
html += "<%= status %>"

details = "<% \n"
details += "jspParameter para = new jspParameter(request); \n"
details += "    int id = para.getInt(\"id\"); \n"
details += "    setPrimaryKey(id) \n"
details += "    findByPrimaryKey() \n"
details += "<table border=0> \n"
for att in table_attnames:
    details += "  <tr><td>"+att+":</td><td><%= "+att+" %></td></tr> \n"
details +="</table> \n"

f=open('html/'+classname+'.html','w')
f.write(MakeJsp(jsp))
f.write(details)
f.write(MakeHtml(html))
f.close