def parse_request(self):
try:
return apiutils.parse_request(self.settings.config.system.secret,
self.request.body)
except Exception as err:
logger.error(u"api authorize parse error, %s" %
utils.safeunicode(traceback.format_exc()))
raise ValueError(u"Error: %s" % utils.safeunicode(err.message))
def info(message, **kwargs):
if not isinstance(message, unicode):
message = safeunicode(message)
if EVENT_INFO in dispatch.dispatch.callbacks:
dispatch.pub(EVENT_INFO, message, **kwargs)
else:
__default_logger_.info(message)
def error(message, **kwargs):
if not isinstance(message, unicode):
message = safeunicode(message)
if EVENT_ERROR in dispatch.dispatch.callbacks:
dispatch.pub(EVENT_ERROR, message, **kwargs)
else:
__default_logger_.error(message)
def debug(message, **kwargs):
if not isinstance(message, unicode):
message = safeunicode(message)
if EVENT_DEBUG in dispatch.dispatch.callbacks:
dispatch.pub(EVENT_DEBUG, message, **kwargs)
else:
__default_logger_.debug(message)
def error(message,**kwargs):
if not isinstance(message, unicode):
message = safeunicode(message)
if EVENT_ERROR in dispatch.dispatch.callbacks:
dispatch.pub(EVENT_ERROR,message,**kwargs)
else:
__default_logger_.error(message)
def post(self):
form = product_forms.product_add_form()
if not form.validates(source=self.get_params()):
return self.render("product_form.html", form=form)
product = models.TrProduct()
product.product_name = form.d.product_name
product.product_policy = form.d.product_policy
product.product_status = form.d.product_status
product.fee_months = int(form.d.get("fee_months", 0))
product.fee_times = utils.hour2sec(form.d.get("fee_times", 0))
product.fee_flows = utils.mb2kb(form.d.get("fee_flows", 0))
product.bind_mac = form.d.bind_mac
product.bind_vlan = form.d.bind_vlan
product.concur_number = form.d.concur_number
product.fee_price = utils.yuan2fen(form.d.fee_price)
product.fee_period = '' #form.d.fee_period or ''
product.input_max_limit = utils.mbps2bps(form.d.input_max_limit)
product.output_max_limit = utils.mbps2bps(form.d.output_max_limit)
_datetime = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
product.create_time = _datetime
product.update_time = _datetime
self.db.add(product)
self.add_oplog(u'新增资费信息:%s' % utils.safeunicode(product.product_name))
self.db.commit()
self.redirect("/admin/product", permanent=False)
def process(self, *args, **kwargs):
with make_db(self.db) as db:
try:
nas_list = db.query(models.TrBas)
for nas in nas_list:
if not nas.dns_name:
continue
results, _, _ = yield client.lookupAddress(nas.dns_name)
if not results:
logger.info("domain {0} resolver empty".format(
nas.dns_name))
if results[0].type == dns.A:
ipaddr = ".".join(
str(i) for i in struct.unpack(
"BBBB", results[0].payload.address))
if ipaddr:
nas.ip_addr = ipaddr
db.commit()
logger.info(
"domain {0} resolver {1} success".format(
nas.dns_name, ipaddr))
else:
logger.info("domain {0} no ip address,{1}".format(
nas.dns_name, repr(results)))
except Exception as err:
logger.error('ddns process error %s' %
utils.safeunicode(err.message))
defer.returnValue(60)
def render_result(self, **result):
resp = apiutils.make_message(self.settings.config.system.secret,
**result)
if self.settings.debug:
logger.debug("[api debug] :: %s response body: %s" %
(self.request.path, utils.safeunicode(resp)))
self.write(resp)
def post(self):
nodes = [(n.node_name, n.node_desc)
for n in self.db.query(models.TrNode)]
products = [(p.id, p.product_name)
for p in self.db.query(models.TrProduct)]
form = operator_form.operator_update_form(nodes, products)
if not form.validates(source=self.get_params()):
rules = self.db.query(models.TrOperatorRule.rule_path).filter_by(
operator_name=form.d.operator_name)
rules = [r[0] for r in rules]
return self.render("base_form.html", form=form, rules=rules)
operator = self.db.query(models.TrOperator).get(form.d.id)
if form.d.operator_pass:
operator.operator_pass = md5(
form.d.operator_pass.encode()).hexdigest()
operator.operator_desc = form.d.operator_desc
operator.operator_status = form.d.operator_status
self.db.query(models.TrOperatorNodes).filter_by(
operator_name=operator.operator_name).delete()
for node in self.get_arguments("operator_nodes"):
onode = models.TrOperatorNodes()
onode.operator_name = form.d.operator_name
onode.node_name = node
self.db.add(onode)
self.db.query(models.TrOperatorProducts).filter_by(
operator_name=operator.operator_name).delete()
for product_id in self.get_arguments("operator_products"):
oproduct = models.TrOperatorProducts()
oproduct.operator_name = form.d.operator_name
oproduct.product_id = product_id
self.db.add(oproduct)
self.add_oplog(u'修改操作员%s信息' %
utils.safeunicode(operator.operator_name))
# update rules
self.db.query(models.TrOperatorRule).filter_by(
operator_name=operator.operator_name).delete()
for path in self.get_arguments("rule_item"):
item = permit.get_route(path)
if not item: continue
rule = models.TrOperatorRule()
rule.operator_name = operator.operator_name
rule.rule_name = item['name']
rule.rule_path = item['path']
rule.rule_category = item['category']
self.db.add(rule)
permit.unbind_opr(operator.operator_name)
self.db.commit()
for rule in self.db.query(models.TrOperatorRule).filter_by(
operator_name=operator.operator_name):
permit.bind_opr(rule.operator_name, rule.rule_path)
self.redirect("/admin/operator", permanent=False)
def debug(message,**kwargs):
if not isinstance(message, unicode):
message = safeunicode(message)
if EVENT_DEBUG in dispatch.dispatch.callbacks:
dispatch.pub(EVENT_DEBUG,message,**kwargs)
else:
__default_logger_.debug(message)
def info(message,**kwargs):
if not isinstance(message, unicode):
message = safeunicode(message)
if EVENT_INFO in dispatch.dispatch.callbacks:
dispatch.pub(EVENT_INFO,message,**kwargs)
else:
__default_logger_.info(message)
def post(self):
try:
req_msg = self.parse_request()
app = self.application
auth = RadiusAuth(app.db_engine, app.mcache, app.aes, req_msg)
self.render_result(**auth.authorize())
except Exception as err:
return self.render_result(code=1,
msg=utils.safeunicode(err.message))
def get(self):
operator_id = self.get_argument("operator_id")
opr = self.db.query(models.TrOperator).get(operator_id)
self.db.query(models.TrOperatorRule).filter_by(
operator_name=opr.operator_name).delete()
self.db.query(models.TrOperator).filter_by(id=operator_id).delete()
self.add_oplog(u'删除操作员%s信息' % utils.safeunicode(opr.operator_name))
self.db.commit()
self.redirect("/admin/operator", permanent=False)
def process(self, *args, **kwargs):
try:
self.statdata.run_stat()
if self.cache.get(radius_statcache_key):
self.cache.update(radius_statcache_key,self.statdata)
else:
self.cache.set(radius_statcache_key,self.statdata)
except Exception as err:
logger.error('radius stat process error %s' % utils.safeunicode(err.message))
return 10.0
def make_sign(api_secret, params=[]):
"""
>>> make_sign("123456",[1,'2',u'中文'])
'33C9065427EECA3490C5642C99165145'
"""
_params = [utils.safeunicode(p) for p in params if p is not None]
_params.sort()
_params.insert(0, api_secret)
strs = ''.join(_params)
mds = md5(strs.encode('utf-8')).hexdigest()
return mds.upper()
def make_sign(api_secret, params=[]):
"""
>>> make_sign("123456",[1,'2',u'中文'])
'33C9065427EECA3490C5642C99165145'
"""
_params = [utils.safeunicode(p) for p in params if p is not None]
_params.sort()
_params.insert(0, api_secret)
strs = ''.join(_params)
mds = md5(strs.encode('utf-8')).hexdigest()
return mds.upper()
def post(self):
@self.cache.cache(expire=60)
def get_bas_by_addr(nasaddr):
return self.db.query(
models.TrBas).filter_by(ip_addr=nasaddr).first()
try:
req_msg = self.parse_request()
if 'nasaddr' not in req_msg:
raise ValueError(u"nasaddr is empty")
except Exception as err:
self.render_result(code=1, msg=utils.safeunicode(err.message))
return
try:
nasaddr = req_msg['nasaddr']
nas = get_bas_by_addr(nasaddr)
if not nas:
self.render_result(code=1,
msg=u'nas {0} not exists'.format(nasaddr))
return
api_addr = "{0}://{1}".format(self.request.protocol,
self.request.host)
result = {
'code': 0,
'msg': 'ok',
'ipaddr': nasaddr,
'secret': nas.bas_secret,
'vendor_id': nas.vendor_id,
'coa_port': int(nas.coa_port or 3799),
'nonce': str(int(time.time())),
}
self.render_result(**result)
except Exception as err:
logger.error(u"api fetch nas error, %s" %
utils.safeunicode(traceback.format_exc()))
self.render_result(code=1, msg=u"api error")
def post(self):
try:
req_msg = self.parse_request()
if req_msg['acct_status_type'] in AcctountingHandler.acct_class:
acctcls = AcctountingHandler.acct_class[
req_msg.acct_status_type]
app = self.application
acctcls(app.db_engine, app.mcache, app.aes,
req_msg).acctounting()
self.render_result(code=0, msg='done')
except Exception as err:
self.render_result(code=1, msg=utils.safeunicode(err))
return
def post(self):
account_number = self.get_argument("account_number")
account = self.db.query(models.TrAccount).get(account_number)
user = self.query_account(account_number)
form = account_forms.account_cancel_form()
if account.status != 1:
return self.render("account_form.html",
user=user,
form=form,
msg=u"无效用户状态")
if not form.validates(source=self.get_params()):
return self.render("account_form.html", user=user, form=form)
accept_log = models.TrAcceptLog()
accept_log.accept_type = 'cancel'
accept_log.accept_source = 'console'
accept_log.account_number = form.d.account_number
accept_log.accept_time = utils.get_currtime()
accept_log.operator_name = self.current_user.username
accept_log.accept_desc = u"用户销户退费%s(元);%s" % (
form.d.fee_value, utils.safeunicode(form.d.operate_desc))
self.db.add(accept_log)
self.db.flush()
self.db.refresh(accept_log)
old_expire_date = account.expire_date
order = models.TrCustomerOrder()
order.order_id = utils.gen_order_id()
order.customer_id = user.customer_id
order.product_id = user.product_id
order.account_number = form.d.account_number
order.order_fee = 0
order.actual_fee = -utils.yuan2fen(form.d.fee_value)
order.pay_status = 1
order.order_source = 'console'
order.accept_id = accept_log.id
order.create_time = utils.get_currtime()
order.order_desc = accept_log.accept_desc
self.db.add(order)
account.status = 3
self.db.commit()
dispatch.pub(ACCOUNT_CHANNEL_EVENT, account.account_number, async=True)
dispatch.pub(db_cache.CACHE_DELETE_EVENT,
account_cache_key(account.account_number),
async=True)
self.redirect(self.detail_url_fmt(account_number))
def processAcct(self, datagram, host, port):
try:
bas = self.find_nas(host)
if not bas:
raise PacketError(
'[Radiusd] :: Dropping packet from unknown host %s' % host)
secret, vendor_id = bas['bas_secret'], bas['vendor_id']
req = self.createAcctPacket(packet=datagram,
dict=self.dict,
secret=six.b(str(secret)),
vendor_id=vendor_id)
self.do_stat(req.code, req.get_acct_status_type())
logger.info("[Radiusd] :: Received radius request: %s" %
(repr(req)))
if self.config.system.debug:
logger.debug(req.format_str())
if req.code != packet.AccountingRequest:
raise PacketError(
'non-AccountingRequest packet on authentication socket')
if not req.VerifyAcctRequest():
raise PacketError('VerifyAcctRequest error')
reply = req.CreateReply()
self.pusher.push(msgpack.packb([reply.ReplyPacket(), host, port]))
self.do_stat(reply.code)
logger.info("[Radiusd] :: Send radius response: %s" % repr(reply))
if self.config.system.debug:
logger.debug(reply.format_str())
status_type = req.get_acct_status_type()
if status_type in self.acct_class:
acct_func = self.acct_class[status_type](
self.db_engine, self.mcache, None,
req.get_ticket()).acctounting
reactor.callLater(0.1, acct_func)
else:
logger.error('status_type <%s> not support' % status_type)
except Exception as err:
self.do_stat(0)
errstr = 'RadiusError:Dropping invalid acct packet from {0} {1},{2}'.format(
host, port, utils.safeunicode(err))
logger.error(errstr)
import traceback
traceback.print_exc()
def post(self):
try:
request = self.parse_form_request()
customer_name = request.get('customer_name')
if not customer_name:
raise Exception("customer_name is empty")
customer = self.db.query(models.TrCustomer).filter_by(
customer_name=customer_name).first()
if not customer:
raise Exception("customer is not exists")
for account in self.db.query(models.TrAccount).filter_by(
customer_id=customer.customer_id):
self.db.query(models.TrAcceptLog).filter_by(
account_number=account.account_number).delete()
self.db.query(models.TrAccountAttr).filter_by(
account_number=account.account_number).delete()
self.db.query(models.TrBilling).filter_by(
account_number=account.account_number).delete()
self.db.query(models.TrTicket).filter_by(
account_number=account.account_number).delete()
self.db.query(models.TrOnline).filter_by(
account_number=account.account_number).delete()
self.db.query(models.TrAccount).filter_by(
account_number=account.account_number).delete()
self.db.query(models.TrCustomerOrder).filter_by(
account_number=account.account_number).delete()
self.add_oplog(u'API删除用户账号%s' % (account.account_number))
dispatch.pub(ACCOUNT_DELETE_EVENT,
account.account_number,
async=True)
dispatch.pub(cache.CACHE_DELETE_EVENT,
account_cache_key(account.account_number),
async=True)
self.db.query(models.TrCustomer).filter_by(
customer_name=customer_name).delete()
self.add_oplog(u'API删除用户资料 %s' % (customer_name))
self.db.commit()
return self.render_result(code=0, msg='success')
except Exception as err:
self.render_result(code=1, msg=utils.safeunicode(err.message))
import traceback
traceback.print_exc()
return
def post(self):
try:
request = self.parse_form_request()
customer_name = request.get('customer_name')
account_number = request.get('account_number')
if not any([customer_name, account_number]):
raise Exception("customer_name,account_number must one")
customer = None
if customer_name:
customer = self.db.query(models.TrCustomer).filter_by(
customer_name=customer_name).first()
else:
customer = self.db.query(models.TrCustomer).filter(
models.TrCustomer.customer_id ==
models.TrAccount.customer_id,
models.TrAccount.account_number == account_number).first()
if not customer:
raise Exception("customer is not exists")
excludes = [
'password', 'email_active', 'active_code', 'mobile_active'
]
customer_data = { c.name : getattr(customer, c.name) \
for c in customer.__table__.columns if c.name not in excludes}
accounts = self.db.query(
models.TrAccount).filter_by(customer_id=customer.customer_id)
if account_number:
accounts = accounts.filter_by(account_number=account_number)
account_datas = []
for account in accounts:
account_data = { c.name : getattr(account, c.name) \
for c in account.__table__.columns if c.name not in 'password'}
account_datas.append(account_data)
self.render_result(code=0,
msg='success',
customer=customer_data,
accounts=account_datas)
except Exception as err:
self.render_result(code=1, msg=utils.safeunicode(err.message))
return
def post(self):
try:
request = self.parse_form_request()
product_id = request.get('product_id')
products = self.db.query(models.TrProduct)
if product_id:
products = products.filter_by(id=product_id)
product_datas = []
excludes = ['fee_period']
for product in products:
product_data = { c.name : getattr(product, c.name) \
for c in product.__table__.columns if c.name not in excludes}
product_datas.append(product_data)
self.render_result(code=0, msg='success', products=product_datas)
except Exception as err:
self.render_result(code=1, msg=utils.safeunicode(err.message))
return
def authorize(self):
try:
if not self.account:
self.failure('user %s not exists'% self.request.account_number)
return self.reply
self.product = self.get_product_by_id(self.account.product_id)
if not self.product:
self.failure('product %s not exists'% self.account.product_id)
return self.reply
for filter_func in self.filters:
flag = filter_func()
if not flag:
return self.reply
return self.reply
except Exception as err:
self.failure("radius authorize error, %s" % utils.safeunicode(err.message))
traceback.print_exc()
return self.reply
def post(self):
try:
request = self.parse_form_request()
node_id = request.get('node_id')
nodes = self.db.query(models.TrNode)
if node_id:
nodes = nodes.filter_by(id=node_id)
node_datas = []
for node in nodes:
node_data = {
c.name: getattr(node, c.name)
for c in node.__table__.columns
}
node_datas.append(node_data)
self.render_result(code=0, msg='success', nodes=node_datas)
except Exception as err:
self.render_result(code=1, msg=utils.safeunicode(err.message))
return
def post(self):
try:
request = self.parse_form_request()
account_number = request.get('account_number')
customer_name = request.get('customer_name')
password = request.get('password')
if not any([account_number, customer_name]):
raise Exception("account_number, customer_name must one")
if not password:
raise Exception("password is empty")
customer, account = None, None
if customer_name:
customer = self.db.query(models.TrCustomer).filter_by(
customer_name=customer_name).first()
if account_number:
account = self.db.query(models.TrAccount).filter_by(
account_number=account_number).first()
if not any([customer, account]):
raise Exception('auth failure,customer or account not exists')
if customer and md5(
password.encode()).hexdigest() == customer.password:
return self.render_result(code=0, msg='success')
if account and password == self.aes.decrypt(account.password):
return self.render_result(code=0, msg='success')
raise Exception('auth failure, password not match')
except Exception as err:
self.render_result(code=1, msg=utils.safeunicode(err.message))
import traceback
traceback.print_exc()
return
def processAuth(self, datagram, host, port):
try:
bas = self.find_nas(host)
if not bas:
raise PacketError(
'[Radiusd] :: Dropping packet from unknown host %s' % host)
secret, vendor_id = bas['bas_secret'], bas['vendor_id']
req = self.createAuthPacket(packet=datagram,
dict=self.dict,
secret=six.b(str(secret)),
vendor_id=vendor_id)
self.do_stat(req.code)
logger.info("[Radiusd] :: Received radius request: %s" %
(repr(req)))
if self.config.system.debug:
logger.debug(req.format_str())
if req.code != packet.AccessRequest:
raise PacketError(
'non-AccessRequest packet on authentication socket')
reply = req.CreateReply()
reply.vendor_id = req.vendor_id
aaa_request = dict(account_number=req.get_user_name(),
domain=req.get_domain(),
macaddr=req.client_mac,
nasaddr=req.get_nas_addr(),
vlanid1=req.vlanid1,
vlanid2=req.vlanid2)
auth_resp = RadiusAuth(self.db_engine, self.mcache, self.aes,
aaa_request).authorize()
if auth_resp['code'] > 0:
reply['Reply-Message'] = auth_resp['msg']
reply.code = packet.AccessReject
return reply
if 'bypass' in auth_resp and int(auth_resp['bypass']) == 0:
is_pwd_ok = True
else:
is_pwd_ok = req.is_valid_pwd(auth_resp.get('passwd'))
if not is_pwd_ok:
reply['Reply-Message'] = "password not match"
reply.code = packet.AccessReject
return reply
else:
if u"input_rate" in auth_resp and u"output_rate" in auth_resp:
reply = rate_process.process(
reply,
input_rate=auth_resp['input_rate'],
output_rate=auth_resp['output_rate'])
attrs = auth_resp.get("attrs") or {}
for attr_name in attrs:
try:
# todo: May have a type matching problem
reply.AddAttribute(utils.safestr(attr_name),
attrs[attr_name])
except Exception as err:
errstr = "RadiusError:current radius cannot support attribute {0},{1}".format(
attr_name, utils.safestr(err.message))
logger.error(errstr)
for attr, attr_val in req.resp_attrs.iteritems():
reply[attr] = attr_val
reply['Reply-Message'] = 'success!'
reply.code = packet.AccessAccept
if not req.VerifyReply(reply):
raise PacketError('VerifyReply error')
return reply
except Exception as err:
self.do_stat(0)
errstr = 'RadiusError:Dropping invalid auth packet from {0} {1},{2}'.format(
host, port, utils.safeunicode(err))
logger.error(errstr)
import traceback
traceback.print_exc()
def post(self):
form = customer_add_vform()
try:
request = self.parse_form_request()
if not form.validates(**request):
raise Exception(form.errors)
if self.db.query(models.TrAccount).filter_by(
account_number=form.d.account_number).count() > 0:
raise Exception("account already exists")
except Exception as err:
self.render_result(code=1, msg=utils.safeunicode(err.message))
return
try:
customer = models.TrCustomer()
customer.node_id = form.d.node_id
customer.realname = form.d.realname
customer.idcard = form.d.idcard
customer.customer_name = form.d.customer_name or form.d.account_number
customer.password = md5(form.d.password.encode()).hexdigest()
customer.sex = '1'
customer.age = '0'
customer.email = form.d.email
customer.mobile = form.d.mobile
customer.address = form.d.address
customer.create_time = form.d.begin_date + ' 00:00:00'
customer.update_time = utils.get_currtime()
customer.email_active = 1
customer.mobile_active = 1
customer.active_code = utils.get_uuid()
self.db.add(customer)
self.db.flush()
self.db.refresh(customer)
accept_log = models.TrAcceptLog()
accept_log.accept_type = 'open'
accept_log.accept_source = 'api'
accept_log.accept_desc = u"API开通账号：%s" % form.d.account_number
accept_log.account_number = form.d.account_number
accept_log.accept_time = customer.update_time
accept_log.operator_name = 'api'
self.db.add(accept_log)
self.db.flush()
self.db.refresh(accept_log)
order_fee = 0
actual_fee = 0
balance = 0
time_length = 0
flow_length = 0
expire_date = form.d.expire_date
product = self.db.query(models.TrProduct).get(form.d.product_id)
# 买断时长
if product.product_policy == BOTimes:
time_length = int(form.d.time_length)
# 买断流量
elif product.product_policy == BOFlows:
flow_length = int(form.d.flow_length)
# 预付费时长,预付费流量
elif product.product_policy in (PPTimes, PPFlow):
balance = utils.yuan2fen(form.d.balance)
expire_date = MAX_EXPIRE_DATE
order = models.TrCustomerOrder()
order.order_id = utils.gen_order_id()
order.customer_id = customer.customer_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 = customer.update_time
order.order_desc = u"API开通账号"
self.db.add(order)
account = models.TrAccount()
account.account_number = form.d.account_number
account.customer_id = customer.customer_id
account.product_id = order.product_id
account.install_address = customer.address
account.ip_address = ''
account.mac_addr = ''
account.password = self.aes.encrypt(form.d.password)
account.status = 1
account.balance = balance
account.time_length = time_length
account.flow_length = flow_length
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_id1 = 0
account.vlan_id2 = 0
account.create_time = customer.create_time
account.update_time = customer.update_time
self.db.add(account)
self.add_oplog(u"API开户，%s" % form.d.account_number)
self.db.commit()
self.render_result(code=0, msg='success')
except Exception as e:
self.render_result(code=1, msg=utils.safeunicode(e.message))
import traceback
traceback.print_exc()
def post(self):
account_number = self.get_argument("account_number")
products = [(n.id, n.product_name) for n in self.get_opr_products()]
form = account_forms.account_change_form(products=products)
account = self.db.query(models.TrAccount).get(account_number)
user = self.query_account(account_number)
if account.status not in (1, 4):
return self.render("account_change_form.html",
user=user,
form=form,
msg=u"无效用户状态")
if not form.validates(source=self.get_params()):
return self.render("account_change_form.html",
user=user,
form=form)
product = self.db.query(models.TrProduct).get(form.d.product_id)
accept_log = models.TrAcceptLog()
accept_log.accept_type = 'change'
accept_log.accept_source = 'console'
accept_log.account_number = form.d.account_number
accept_log.accept_time = utils.get_currtime()
accept_log.operator_name = self.current_user.username
accept_log.accept_desc = u"用户资费变更为:%s;%s" % (
product.product_name, utils.safeunicode(form.d.operate_desc))
self.db.add(accept_log)
self.db.flush()
self.db.refresh(accept_log)
old_exoire_date = account.expire_date
account.product_id = product.id
# (PPMonth,PPTimes,BOMonth,BOTimes,PPFlow,BOFlows)
if product.product_policy in (PPMonth, BOMonth):
account.expire_date = form.d.expire_date
account.balance = 0
account.time_length = 0
account.flow_length = 0
elif product.product_policy in (PPTimes, PPFlow):
account.expire_date = MAX_EXPIRE_DATE
account.balance = utils.yuan2fen(form.d.balance)
account.time_length = 0
account.flow_length = 0
elif product.product_policy == BOTimes:
account.expire_date = MAX_EXPIRE_DATE
account.balance = 0
account.time_length = utils.hour2sec(form.d.time_length)
account.flow_length = 0
elif product.product_policy == BOFlows:
account.expire_date = MAX_EXPIRE_DATE
account.balance = 0
account.time_length = 0
account.flow_length = utils.mb2kb(form.d.flow_length)
order = models.TrCustomerOrder()
order.order_id = utils.gen_order_id()
order.customer_id = account.customer_id
order.product_id = account.product_id
order.account_number = account.account_number
order.order_fee = 0
order.actual_fee = utils.yuan2fen(form.d.add_value) - utils.yuan2fen(
form.d.back_value)
order.pay_status = 1
order.accept_id = accept_log.id
order.order_source = 'console'
order.create_time = utils.get_currtime()
order.order_desc = u"用户变更资费,变更前到期:%s,变更后到期:%s" % (old_exoire_date,
account.expire_date)
self.db.add(order)
self.add_oplog(accept_log.accept_desc)
self.db.commit()
dispatch.pub(db_cache.CACHE_DELETE_EVENT,
account_cache_key(account.account_number),
async=True)
self.redirect(self.detail_url_fmt(account_number))
def post(self):
nodes = [(n.id, n.node_desc) for n in self.get_opr_nodes()]
products = [(n.id, n.product_name) for n in self.get_opr_products()]
iform = customer_forms.customer_import_form(nodes, products)
node_id = self.get_argument('node_id')
product_id = self.get_argument('product_id')
f = self.request.files['import_file'][0]
impctx = utils.safeunicode(f['body'])
lines = impctx.split("\n")
_num = 0
impusers = []
for line in lines:
_num += 1
line = line.strip()
if not line or u"用户姓名" in line: continue
attr_array = line.split(",")
if len(attr_array) < 11:
return self.render("customer_import_form.html",
form=iform,
msg=u"line %s error: length must 11 " %
_num)
vform = customer_forms.customer_import_vform()
if not vform.validates(
dict(realname=attr_array[0],
idcard=attr_array[1],
mobile=attr_array[2],
address=attr_array[3],
account_number=attr_array[4],
password=attr_array[5],
begin_date=attr_array[6],
expire_date=attr_array[7],
balance=attr_array[8],
time_length=utils.hour2sec(attr_array[9]),
flow_length=utils.mb2kb(attr_array[10]))):
return self.render("customer_import_form.html",
form=iform,
msg=u"line %s error: %s" %
(_num, vform.errors))
impusers.append(vform)
_unums = 0
for form in impusers:
try:
customer = models.TrCustomer()
customer.node_id = node_id
customer.realname = form.d.realname
customer.idcard = form.d.idcard
customer.customer_name = form.d.account_number
customer.password = md5(form.d.password.encode()).hexdigest()
customer.sex = '1'
customer.age = '0'
customer.email = ''
customer.mobile = form.d.mobile
customer.address = form.d.address
customer.create_time = form.d.begin_date + ' 00:00:00'
customer.update_time = utils.get_currtime()
customer.email_active = 0
customer.mobile_active = 0
customer.active_code = utils.get_uuid()
self.db.add(customer)
self.db.flush()
self.db.refresh(customer)
accept_log = models.TrAcceptLog()
accept_log.accept_type = 'open'
accept_log.accept_source = 'console'
_desc = u"用户导入账号：%s" % form.d.account_number
accept_log.accept_desc = _desc
accept_log.account_number = form.d.account_number
accept_log.accept_time = customer.update_time
accept_log.operator_name = self.current_user.username
self.db.add(accept_log)
self.db.flush()
self.db.refresh(accept_log)
order_fee = 0
actual_fee = 0
balance = 0
time_length = 0
flow_length = 0
expire_date = form.d.expire_date
product = self.db.query(models.TrProduct).get(product_id)
# 买断时长
if product.product_policy == BOTimes:
time_length = int(form.d.time_length)
# 买断流量
elif product.product_policy == BOFlows:
flow_length = int(form.d.flow_length)
# 预付费时长,预付费流量
elif product.product_policy in (PPTimes, PPFlow):
balance = utils.yuan2fen(form.d.balance)
expire_date = MAX_EXPIRE_DATE
order = models.TrCustomerOrder()
order.order_id = utils.gen_order_id()
order.customer_id = customer.customer_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 = customer.update_time
order.order_desc = u"用户导入开户"
self.db.add(order)
account = models.TrAccount()
account.account_number = form.d.account_number
account.customer_id = customer.customer_id
account.product_id = order.product_id
account.install_address = customer.address
account.ip_address = ''
account.mac_addr = ''
account.password = self.aes.encrypt(form.d.password)
account.status = 1
account.balance = balance
account.time_length = time_length
account.flow_length = flow_length
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_id1 = 0
account.vlan_id2 = 0
account.create_time = customer.create_time
account.update_time = customer.update_time
self.db.add(account)
_unums += 1
except Exception as e:
return self.render("customer_import_form.html",
form=iform,
msg=u"error : %s" % str(e))
self.add_oplog(u"导入开户，用户数：%s" % _unums)
self.db.commit()
self.redirect("/admin/customer")
