diff options
Diffstat (limited to 'server/piztor/server.py')
-rw-r--r-- | server/piztor/server.py | 12 |
1 files changed, 9 insertions, 3 deletions
diff --git a/server/piztor/server.py b/server/piztor/server.py index dab0ccb..78a15b6 100644 --- a/server/piztor/server.py +++ b/server/piztor/server.py @@ -27,6 +27,10 @@ logger = logging.getLogger('piztor_server') logger.setLevel(logging.INFO) engine = create_engine('mysql://' + db_path, echo = False, pool_size = 1024) +class _PermCode: + normal = 0x00 + sectoin = 0x01 + company = 0x02 class _SectionSize: LENGTH = 4 @@ -596,9 +600,9 @@ class SetMarkerHandler(RequestHandler): if username is None: raise struct.error except struct.error: - raise BadReqError("Send text mesg: Malformed request body") + raise BadReqError("Set marker: Malformed request body") - logger.info("Trying to send text mesg with " + logger.info("Trying to set marker with " "(token = {0}, username = {1})"\ .format(get_hex(token), username)) @@ -610,8 +614,10 @@ class SetMarkerHandler(RequestHandler): pt = RequestHandler.push_tunnels u = uauth.user - ulist = self.session.query(UserModel) \ + if u.perm == _PermCode.section: + ulist = self.session.query(UserModel) \ .filter(UserModel.sec_id == u.sec_id).all() + else if u.perm == _PermCode.section: for user in ulist: uid = user.id |