def member_join_post(db):
nodes = [(n.id, n.node_name) for n in db.query(models.SlcNode)]
form = forms.member_join_form(nodes)
if not form.validates(source=request.params):
return render("join", form=form)
if db.query(exists().where(
models.SlcMember.member_name == form.d.username)).scalar():
return render("join",
form=form,
msg=u"用户{0}已被使用".format(form.d.username))
if db.query(
exists().where(models.SlcMember.email == form.d.email)).scalar():
return render("join",
form=form,
msg=u"用户邮箱{0}已被使用".format(form.d.email))
member = models.SlcMember()
member.node_id = form.d.node_id
member.realname = form.d.realname
member.member_name = form.d.username
member.password = md5(form.d.password.encode()).hexdigest()
member.idcard = form.d.idcard
member.sex = form.d.sex
member.age = int(form.d.age)
member.email = form.d.email
member.mobile = form.d.mobile
member.address = form.d.address
member.create_time = utils.get_currtime()
member.update_time = utils.get_currtime()
db.add(member)
db.commit()
logger.info(u"新用户注册成功,member_name=%s" % member.member_name)
redirect('/login')
def member_import(db):
nodes = [(n.id, n.node_name) for n in db.query(models.SlcNode)]
products = [(p.id, p.product_name) for p in db.query(models.SlcRadProduct)]
iform = forms.user_import_form(nodes, products)
node_id = request.params.get('node_id')
product_id = request.params.get('product_id')
upload = request.files.get('import_file')
impctx = upload.file.read()
lines = impctx.split("\n")
_num = 0
impusers = []
for line in lines:
_num += 1
line = line.strip()
if not line or "用户姓名" in line: continue
attr_array = line.split(",")
if len(attr_array) < 5:
return render("bus_import_form",
form=iform,
msg=u"line %s error: length must 5 " % _num)
vform = forms.user_import_vform()
if not vform.validates(
dict(realname=attr_array[0],
account_number=attr_array[1],
password=attr_array[2],
expire_date=attr_array[3],
balance=attr_array[4])):
return render("bus_import_form",
form=iform,
msg=u"line %s error: %s" % (_num, vform.errors))
impusers.append(vform)
for form in impusers:
try:
member = models.SlcMember()
member.node_id = node_id
member.realname = form.d.realname
member.idcard = '123456'
member.member_name = form.d.account_number
member.password = form.d.account_number
member.sex = '1'
member.age = '0'
member.email = ''
member.mobile = '123456'
member.address = 'address'
member.create_time = utils.get_currtime()
member.update_time = utils.get_currtime()
db.add(member)
db.flush()
db.refresh(member)
accept_log = models.SlcRadAcceptLog()
accept_log.accept_type = 'open'
accept_log.accept_source = 'console'
accept_log.accept_desc = u"用户导入账号：(%s)%s - 上网账号:%s" % (
member.member_name, member.realname, form.d.account_number)
accept_log.account_number = form.d.account_number
accept_log.accept_time = member.create_time
accept_log.operator_name = get_cookie("username")
db.add(accept_log)
db.flush()
db.refresh(accept_log)
order_fee = 0
actual_fee = 0
balance = 0
expire_date = form.d.expire_date
product = db.query(models.SlcRadProduct).get(product_id)
if product.product_policy == 1:
balance = int(form.d.balance)
expire_date = '3000-11-11'
order = models.SlcMemberOrder()
order.order_id = utils.gen_order_id()
order.member_id = member.member_id
order.product_id = product.id
order.account_number = form.d.account_number
order.order_fee = order_fee
order.actual_fee = actual_fee
order.pay_status = 1
order.accept_id = accept_log.id
order.order_source = 'console'
order.create_time = member.create_time
order.order_desc = u"用户导入开户"
db.add(order)
account = models.SlcRadAccount()
account.account_number = form.d.account_number
account.member_id = member.member_id
account.product_id = order.product_id
account.install_address = member.address
account.ip_address = ''
account.mac_addr = ''
account.password = utils.encrypt(form.d.password)
account.status = 1
account.balance = balance
account.time_length = 0
account.expire_date = expire_date
account.user_concur_number = product.concur_number
account.bind_mac = product.bind_mac
account.bind_vlan = product.bind_vlan
account.vlan_id = 0
account.vlan_id2 = 0
account.create_time = member.create_time
account.update_time = member.create_time
db.add(account)
except Exception as e:
return render("bus_import_form",
form=iform,
msg=u"error : %s" % str(e))
db.commit()
redirect("/bus/member")
def member_open(db):
nodes = [(n.id, n.node_name) for n in db.query(models.SlcNode)]
products = [
(n.id, n.product_name)
for n in db.query(models.SlcRadProduct).filter_by(product_status=0)
]
form = forms.user_open_form(nodes, products)
if not form.validates(source=request.forms):
return render("bus_open_form", form=form)
if db.query(models.SlcRadAccount).filter_by(
account_number=form.d.account_number).count() > 0:
return render("bus_open_form",
form=form,
msg=u"上网账号%s已经存在" % form.d.account_number)
if form.d.ip_address and db.query(models.SlcRadAccount).filter_by(
ip_address=form.d.ip_address).count() > 0:
return render("bus_open_form",
form=form,
msg=u"ip%s已经被使用" % form.d.ip_address)
if db.query(models.SlcMember).filter_by(
member_name=form.d.member_name).count() > 0:
return render("bus_open_form",
form=form,
msg=u"用户名%s已经存在" % form.d.member_name)
member = models.SlcMember()
member.node_id = form.d.node_id
member.realname = form.d.realname
member.member_name = form.d.member_name
member.password = md5(form.d.member_password.encode()).hexdigest()
member.idcard = form.d.idcard
member.sex = '1'
member.age = '0'
member.email = ''
member.mobile = form.d.mobile
member.address = form.d.address
member.create_time = utils.get_currtime()
member.update_time = utils.get_currtime()
db.add(member)
db.flush()
db.refresh(member)
accept_log = models.SlcRadAcceptLog()
accept_log.accept_type = 'open'
accept_log.accept_source = 'console'
accept_log.account_number = form.d.account_number
accept_log.accept_time = member.create_time
accept_log.operator_name = get_cookie("username")
accept_log.accept_desc = u"用户新开户：(%s)%s - 上网账号:%s" % (
member.member_name, member.realname, form.d.account_number)
db.add(accept_log)
db.flush()
db.refresh(accept_log)
order_fee = 0
balance = 0
expire_date = form.d.expire_date
product = db.query(models.SlcRadProduct).get(form.d.product_id)
if product.product_policy == 0:
order_fee = decimal.Decimal(product.fee_price) * decimal.Decimal(
form.d.months)
order_fee = int(order_fee.to_integral_value())
elif product.product_policy == 1:
balance = utils.yuan2fen(form.d.fee_value)
expire_date = '3000-11-11'
order = models.SlcMemberOrder()
order.order_id = utils.gen_order_id()
order.member_id = member.member_id
order.product_id = product.id
order.account_number = form.d.account_number
order.order_fee = order_fee
order.actual_fee = utils.yuan2fen(form.d.fee_value)
order.pay_status = 1
order.accept_id = accept_log.id
order.order_source = 'console'
order.create_time = member.create_time
order.order_desc = u"用户新开账号"
db.add(order)
account = models.SlcRadAccount()
account.account_number = form.d.account_number
account.ip_address = form.d.ip_address
account.member_id = member.member_id
account.product_id = order.product_id
account.install_address = member.address
account.mac_addr = ''
account.password = utils.encrypt(form.d.password)
account.status = form.d.status
account.balance = balance
account.time_length = 0
account.expire_date = expire_date
account.user_concur_number = product.concur_number
account.bind_mac = product.bind_mac
account.bind_vlan = product.bind_vlan
account.vlan_id = 0
account.vlan_id2 = 0
account.create_time = member.create_time
account.update_time = member.create_time
db.add(account)
db.commit()
redirect("/bus/member")