def add_acct():
print (str(session['email']),'is on add_acct')
form = AccountForm()
if request.method == 'POST':
if form.validate() == False:
return render_template('addacct.html', form=form)
else:
newacct = Accounts(form.name.data, session['email'])
db_session.add(newacct)
db_session.commit()
flash('You created a new account.')
print (str(session['email']),'has successfully added a new account')
return redirect(url_for('home/'))
elif request.method == 'GET':
print (str(session['email']),'is still on add_acct')
return render_template('addacct.html', form=form)
def create_user():
form = AccountForm(request.form)
if form.validate():
user = models.get_user(form.data['username'])
if user != None:
message = "User already exists. Please choose another username."
return render_template("create.html", form=form, message=message)
elif form.data['password'] != form.data['auth_pass']:
message = "Passwords do not match."
return render_template("create.html", form=form, message=message)
else:
new_user = User(name=form.data['username'],
password=form.data['password'],
role="user")
g.db.add(new_user)
login_user(new_user)
return redirect(url_for("home"))
print form.data
return render_template("create.html", form=form, message=None)
def view_account():
account_id = request.args.get('id')
q = request.args.get('q')
if q is not None:
return redirect(url_for('view_accounts', page=1, q=q))
form = AccountForm(request.form)
if request.method == 'POST' and form.validate():
if form.id.data:
account = orm.Account.query.get(int(form.id.data))
account.username = form.telephone.data
account.name = form.telephone.data
account.telephone = form.telephone.data
account.role = 0
account.flag_telephone = 1 if form.flag_telephone.data else 0
account.checkcode = form.checkcode.data
account.source = form.source.data
account.dtcreate = form.dtcreate.data
orm.db.session.commit()
else:
account = orm.Account(form.telephone.data, '1234',
form.telephone.data, form.telephone.data, 0,
1 if form.flag_telephone.data else 0, '1234',
form.source.data, form.dtcreate.data)
orm.db.session.add(account)
orm.db.session.commit()
form.id.data = account.id
return redirect(url_for('view_account'))
elif request.method == 'GET' and account_id:
form = logic.GetAccountFormById(account_id)
logic.LoadBasePageInfo('修改用户', '输入并确定', form)
else:
logic.LoadBasePageInfo('新建用户', '输入并确定', form)
if form.id.data:
account = orm.Account.query.get(int(form.id.data))
form.account = account
return render_template('view_account.html', form=form)
def view_account():
account_id = request.args.get('id')
q = request.args.get('q')
if q is not None:
return redirect(url_for('view_accounts', page=1, q=q))
form = AccountForm(request.form)
if request.method == 'POST' and form.validate():
if form.id.data:
account = orm.Account.query.get(int(form.id.data))
account.username = form.telephone.data
account.name = form.telephone.data
account.telephone = form.telephone.data
account.role = 0
account.flag_telephone = 1 if form.flag_telephone.data else 0
account.checkcode = form.checkcode.data
account.source = form.source.data
account.dtcreate = form.dtcreate.data
orm.db.session.commit()
else:
account = orm.Account(form.telephone.data, '1234', form.telephone.data, form.telephone.data, 0, 1 if form.flag_telephone.data else 0, '1234', form.source.data, form.dtcreate.data)
orm.db.session.add(account)
orm.db.session.commit()
form.id.data = account.id
return redirect(url_for('view_account'))
elif request.method =='GET' and account_id:
form = logic.GetAccountFormById(account_id)
logic.LoadBasePageInfo('修改用户','输入并确定',form)
else:
logic.LoadBasePageInfo('新建用户','输入并确定',form)
if form.id.data:
account = orm.Account.query.get(int(form.id.data))
form.account = account
return render_template('view_account.html',form = form)
def index(id):
is_account_popup_active = ""
accounts_form = AccountForm(request.form)
accounts_form.hdn_account_id.data = id
if id != 0 and request.method != "POST":
db_account = Account.get_accounts(id)
accounts_form.hdn_account_id.data = db_account[0]["ID"]
accounts_form.account_type.process_data(db_account[0]["TYPE"])
accounts_form.full_address.data = db_account[0][
"ACCOUNT_HOLDER_ADDRESS"]
accounts_form.full_name.data = db_account[0]["NAME"]
accounts_form.mobile.data = db_account[0][
"ACCOUNT_HOLDER_CONTACT_NUMBER"]
accounts_form.account_number.data = db_account[0][
"ACCOUNT_HOLDER_BANK_DETAILS"]
is_account_popup_active = "active"
transaction_form = CreditDebitForm()
transaction_form.FROM_ACCOUNT.choices = Account.get_account_by_type(1)
transaction_form.TO_ACCOUNT.choices = Account.get_account_by_type(2)
print(request.method)
if request.method == "POST":
if accounts_form.validate():
Account.create_account(accounts_form, id)
return redirect(url_for("index", id=0))
else:
is_account_popup_active = "active"
return render_template(
"index.html",
accounts=Account.get_accounts(0),
account_transactions=Account.get_account_transaction(0, 0, 0),
form=accounts_form,
form2=transaction_form,
is_account_popup_active=is_account_popup_active,
is_trans_popop_active="",
)
def account():
if request.method == "GET":
form = AccountForm(obj=current_user)
else:
form = AccountForm(request.form)
try:
subscribed = Subscription.objects.get(user=current_user.pk)
except Subscription.DoesNotExist:
subscribed = None
if not request.method == 'POST' or not form.validate():
context = {'form': form, 'subscribed': subscribed}
return render_template('account.html', **context)
if form.username.data: current_user.username = form.username.data
if form.email.data: current_user.email = form.email.data
if form.phone.data: current_user.phone = form.phone.data
if form.address.data: current_user.address = form.address.data
if form.subscribe.data: current_user.subscribe = form.subscribe.data
if form.password.data: current_user.set_password(form.password.data)
try:
current_user.save()
except Exception, e:
print str(e)
