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 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")