コード例 #1
0
    def post(self):
        uploads = self.get_uploads()
        for file in uploads:
            filename = blobstore.BlobInfo(file.key()).filename

            user = users.get_current_user()
            my_user_key = ndb.Key(MyUser, user.user_id())
            my_user = my_user_key.get()

            currDirObj = my_user.currDir.get()
            path = ''
            if functions.isRootDirectory(my_user):
                path = currDirObj.dirPath + filename
            else:
                path = currDirObj.dirPath + '/' + filename

            file_id = my_user.key.id() + path
            file_key = ndb.Key(File, file_id)
            logging.info(file_id)
            if functions.exists(file_key, currDirObj.files):
                file_object = File(id=file_id)
                file_object.fileName = filename
                file_object.data = file.key()
                file_object.put()

                currDirObj.files.append(file_key)
                currDirObj.put()

            else:
                # Delete uploaded file from the blobstore
                blobstore.delete(file.key())
                logging.debug("File with the same name already exists!")

        self.redirect('/')
コード例 #2
0
ファイル: main.py プロジェクト: sneha2208/Dropbox-Replica
    def post(self):
        self.response.headers['Content-Type'] = 'text/html'
        button = self.request.get('button')
        if button == 'Add':
            directory_name = self.request.get('value')
            directory_name = re.sub(r'[/;]', '', directory_name).lstrip()
            if not (directory_name is None or directory_name == ''):
                user = users.get_current_user()
                my_user_key = ndb.Key(MyUser, user.user_id())
                my_user = my_user_key.get()
                logging.info("my user key in post method")
                logging.info(my_user)
                parent_dir_obj = my_user.currDir.get()

                path = ''
                if functions.isRootDirectory(my_user):
                    path = parent_dir_obj.dirPath + directory_name
                else:
                    path = parent_dir_obj.dirPath + '/' + directory_name

                directory_id = my_user.key.id() + path
                directory = Directory(id=directory_id)
                logging.info("dir ID")
                logging.info(directory_id)
                logging.info(parent_dir_obj.subDir)
                # If directory already exists
                if functions.exists(directory.key, parent_dir_obj.subDir):
                    parent_dir_obj.subDir.append(directory.key)
                    parent_dir_obj.put()
                    logging.info("parent_dir_obj")
                    logging.info(parent_dir_obj)
                    # Saving all params of dir to db
                    directory.parentDirectory = my_user.currDir
                    directory.dirName = directory_name
                    directory.dirPath = path
                    directory.put()
                self.redirect('/')

        elif button == 'Delete':
            name = self.request.get('name')
            type = self.request.get('fType')

            if type == 'file':
                user = users.get_current_user()
                my_user_key = ndb.Key(MyUser, user.user_id())
                my_user = my_user_key.get()
                parent_dir_obj = my_user.currDir.get()
                path = ''
                if functions.isRootDirectory(my_user):
                    path = parent_dir_obj.dirPath + name
                else:
                    path = parent_dir_obj.dirPath + '/' + name
                file_id = my_user.key.id() + path
                file_key = ndb.Key(File, file_id)
                parent_dir_obj.files.remove(file_key)
                parent_dir_obj.put()
                blobObject = file_key.get().data
                blobstore.delete(blobObject)
                file_key.delete()



            elif type == 'directory':
                user = users.get_current_user()
                my_user_key = ndb.Key(MyUser, user.user_id())
                my_user = my_user_key.get()

                parent_dir_obj = my_user.currDir.get()

                path = ''
                if functions.isRootDirectory(my_user):
                    path = parent_dir_obj.dirPath + name
                else:
                    path = parent_dir_obj.dirPath + '/' + name

                directory_id = my_user.key.id() + path
                directory_key = ndb.Key(Directory, directory_id)
                directory_object = directory_key.get()

                if not directory_object.files and not directory_object.subDir:
                    #Delete all the references
                    parent_dir_obj.subDir.remove(directory_key)
                    parent_dir_obj.put()

                    # Delete directory from datastore
                    directory_key.delete()


            self.redirect('/')

        elif button == 'Up':
            user = users.get_current_user()
            my_user_key = ndb.Key(MyUser, user.user_id())
            my_user = my_user_key.get()
            if not functions.isRootDirectory(my_user):
                parent_dir_obj = my_user.currDir.get().parentDirectory
                my_user.currDir = parent_dir_obj
                my_user.put()
            self.redirect('/')

        elif button == 'Home':
            user = users.get_current_user()
            my_user_key = ndb.Key(MyUser, user.user_id())
            my_user = my_user_key.get()
            my_user.currDir = ndb.Key(Directory, my_user.key.id() + '/')
            my_user.put()
            self.redirect('/')
コード例 #3
0
def model_generation():
    input=open("models_char.txt","r")
    output=open("humans.txt","w")
    output2=open("objects.txt","w")
    output3=open("environments.txt","w")
    lines=input.readlines()
    check=False
    for temp in lines:
        line=temp.split()
        human,color,scale=functions.human_path(line[0],line[1],line[2])
        if human!=None:
            if(color=='none'):
                color='NA'
            else:
                color=colors.to_rgba(color)
                color = '_'.join(str(e) for e in color)
            scale3=1
            if(line[3]==0):
                scale3=0.9
            elif line[3]==2:
                scale3=1.2
            scale=str(scale*5)+'_'+str(scale3*scale*5)+'_'+str(scale*5)
            output.write(human+' '+color+' '+scale+' '+str(line[4])+' 0 1\n')
        else :
            if functions.exists(line[0],True):
                size=1
                print(line[2])
                if (line[2]==2):
                    size=size*1.5
                elif line[2]==0:
                    size = size * .75
                if line[1]=='none':
                    color='NA'
                else:
                    color=colors.to_rgba(line[1])
                    color='_'.join(str(e) for e in color)
                size2=size3=size
                if line[0]=='chair' and size2==1.5:
                    size=1
                    size3=1
                    size2=1.2
                output2.write(line[0]+' '+color+' '+str(size2)+'_'+str(size)+'_'+str(+size3)+' '+str(line[3])+' '+str(functions.inv_math[functions.models.index(line[0])])+' '+
                str(functions.e[functions.models.index(line[0])])+'\n')
            elif functions.exists(line[0],False):
                check=True
                output3.write(str(line[0])+' '+str(line[3])+' 0 1')
            else :
                size = 1
                print(line[2])
                if (line[2] == 2):
                    size = size * 1.15
                elif line[2] == 0:
                    size = size * .75
                if line[1] == 'none':
                    color = 'NA'
                else:
                    color = colors.to_rgba(line[1])
                    color = '_'.join(str(e) for e in color)
                output2.write('box' + ' ' + color + ' ' + str(size) + str(line[3]) + ' 0.1 0.1\n')
    if check ==False:
        output3.write('Room 0 0 1\n')
    output.close()
    output2.close()
    output3.close()
    input.close()
コード例 #4
0
def main():
    functions.clear()
    if not functions.exists():
        try:
            functions.intro()
        except KeyboardInterrupt:
            functions.clear()
            import sys
            sys.exit(0)
    functions.clear()
    try:
        password = getpass.getpass("Password: "******"Invalid password")
        import sys
        sys.exit(1)
    while True:
        functions.clear()
        print("Vault")
        print(
            "Use Ctrl+C to return back to this menu, or use it here to exit.\n"
        )
        try:
            action = input(
                "1: Encrypt a file\n2: Download and encrypt a file from a URL\n3: Decrypt a file\n4: Change password\nSelect an option: "
            )
        except:
            break
        action = action.strip().lstrip()
        try:
            action = int(action)
        except:
            continue
        if action not in [1, 2, 3, 4]:
            continue
        if action == 1:
            functions.clear()
            try:
                path = input("Please specify a file to encrypt: ")
            except KeyboardInterrupt:
                continue
            if functions.validate(path) and not path.endswith(".png"):
                functions.clear()
                try:
                    response = input(
                        "This file may be a valid image file. Convert to .png (y/n)? "
                    )
                except KeyboardInterrupt:
                    continue
                if response.lower().lstrip().strip() == "y":
                    temp = path
                    path = functions.convert(path)
                    delete = input("Delete original file (y/n)? ")
                    if delete.lower().lstrip().strip() == "y":
                        if not windows:
                            temp = temp.replace("\\", "")
                        try:
                            directory = os.path.dirname(os.path.realpath(temp))
                            os.chdir(directory)
                            os.remove(temp)
                        except:
                            pass
            functions.clear()
            try:
                save = functions.encodepath(path)
                if save == None:
                    continue
            except Exception as e:
                print(str(e))
                try:
                    input("Invalid file path. Press enter to continue. ")
                except KeyboardInterrupt:
                    continue
                continue
            else:
                try:
                    input(
                        "File encrypted.\nEncrypted file path: {}\nPress enter to continue. "
                        .format(save))
                except KeyboardInterrupt:
                    continue
                continue

        elif action == 2:
            functions.clear()
            try:
                url = input("Please specify a file URL to download: ")
            except KeyboardInterrupt:
                continue

            name = os.path.split(url)[-1]
            base = os.path.dirname(os.path.realpath(__file__))
            path = os.path.join(base, name)
            functions.clear()

            try:
                r = requests.get(url, stream=True)
                with open(path, 'wb') as out_file:
                    shutil.copyfileobj(r.raw, out_file)
            except:
                try:
                    input("Invalid URL: {}\nPress enter to continue. ".format(
                        url))
                except KeyboardInterrupt:
                    continue
                continue
            if functions.validate(path) and not path.endswith(".png"):
                functions.clear()
                try:
                    response = input(
                        "This file may be a valid image file. Convert to .png (y/n)? "
                    )
                except KeyboardInterrupt:
                    continue
                if response.lower().lstrip().strip() == "y":
                    temp = path
                    path = functions.convert(path)
                    delete = input("Delete original file (y/n)? ")
                    if delete.lower().lstrip().strip() == "y":
                        if not windows:
                            temp = temp.replace("\\", "")
                        try:
                            directory = os.path.dirname(os.path.realpath(temp))
                            os.chdir(directory)
                            os.remove(temp)
                        except:
                            pass
            functions.clear()
            try:
                save = functions.encodepath(path)
            except:
                try:
                    input("Invalid file path. Press enter to continue. ")
                except KeyboardInterrupt:
                    continue
                continue
            else:
                try:
                    input(
                        "File encrypted.\nEncrypted file path: {}\nPress enter to continue. "
                        .format(save))
                except KeyboardInterrupt:
                    continue
                continue
        elif action == 3:
            functions.clear()
            try:
                path = input("Please specify a .enc file to decrypt: ")
            except KeyboardInterrupt:
                continue
            functions.clear()
            try:
                save = functions.decodepath(path)
            except Exception as e:
                print(str(e))
                try:
                    input("Invalid file path. Press enter to continue. ")
                except KeyboardInterrupt:
                    continue
                continue
            else:
                try:
                    input(
                        "File decrypted.\nDecrypted file path: {}\nPress enter to continue. "
                        .format(save))
                except KeyboardInterrupt:
                    continue
                continue
        elif action == 4:
            try:
                functions.clear()
                current = input("Current password: "******"Passwords do not match. Press enter to continue. ")
                    continue
                new = input("New password: "******"You will need to decrypt all files with your old password. Your new password will not be able to decrypt already encrypted files.\nRetype your password to confirm: "
                )
                if confirm != new:
                    input("Passwords do not match. Press enter to continue. ")
                    continue
                functions.writepw(new)
                functions.clear()
                input("Password changed to: {}.\nPress enter to continue. ".
                      format(new))
            except KeyboardInterrupt:
                continue

    functions.clear()
コード例 #5
0
def update(bake=False):
    def namespaced(n):
        return '%s:%s' % (name_space, n)

    cmds.currentUnit(time='pal')
    if not f.exists(scene_check_joints):
        exit_message = 'Not all joints from %s present in scene' % scene_check_joints
        sys.exit(exit_message)
    # Imports file in scene with a given name space
    import_file = cmds.fileDialog2(dialogStyle=2, fm=1)[0]
    name_space = os.path.basename(import_file).split('.')[0] + '_namespace'
    cmds.file(import_file, i=True, namespace=name_space)
    # Delete animation from joints in Biped hierarchy
    cmds.currentTime(0)
    cmds.select(f.get_joints())
    cmds.cutKey()
    # Exports skin cluster weights for body meshes in scene
    all_meshes = f.get_meshes()
    head_meshes = []
    search_pattern_r = '|'.join([sp for sp in search_pattern.split('\n')])
    regex = r"\b.*(%s).*\b" % search_pattern_r

    for m in all_meshes:
        if re.search(regex, m):
            head_meshes.append(m)

    body_meshes = [m for m in all_meshes if m not in head_meshes]
    if body_meshes:
        cmds.select(body_meshes)
        sie.export_weights_sp()

    for sc in cmds.ls(type='skinCluster'):
        cmds.skinCluster(sc, ub=True, edit=True)

    # Deletes bind poses in scene
    cmds.delete(cmds.ls(type='dagPose'))
    # Checks if scene joints are scaled
    length = f.get_length(f.get_pos(length_joints[0]),
                          f.get_pos(length_joints[1]))
    imported_length = f.get_length(f.get_pos(namespaced(length_joints[0])),
                                   f.get_pos(namespaced(length_joints[1])))
    scale_factor = length / imported_length
    if round(scale_factor, 0) != 1:
        scaled_joints_group = cmds.group(namespaced(root_joint))
        cmds.xform(scaled_joints_group,
                   scale=(scale_factor, scale_factor, scale_factor))
    # Aligns imported skeleton to the one in scene
    # All created constraints will be deleted alongside with imported mesh later
    cmds.pointConstraint(head_joint, namespaced(head_joint))
    head_children_joints = cmds.listRelatives(head_joint,
                                              ad=True,
                                              type='joint')
    head_children_joints = filter_nubs(head_children_joints)

    for j in head_children_joints:
        if f.exists(namespaced(j)):
            f.unlock_attributes(j)
            cmds.pointConstraint(namespaced(j), j)
            cmds.orientConstraint(namespaced(j), j)

    # Keys all biped joints to prevent joints shift
    if bake:
        # Gets information about keyed range of imported bones
        ns_biped = [
            namespaced(j) for j in f.get_joints() if f.exists(namespaced(j))
        ]
        imported_biped_keys = cmds.keyframe(ns_biped,
                                            query=True,
                                            timeChange=True)
        ns_min, ns_max = min(imported_biped_keys), max(imported_biped_keys)
        cmds.bakeResults(f.get_joints(), time=(ns_min, ns_max + 1), sm=True)
        cmds.playbackOptions(min=ns_min, max=ns_max + 1)
    else:
        cmds.setKeyframe(f.get_joints(), time=0)
    cmds.namespace(removeNamespace=name_space, deleteNamespaceContent=True)
    # Imports all skin cluster weights for meshes in scene
    cmds.select(f.get_meshes())
    sie.import_weights_sp()
    # Fix eyes joints pivot.
    # When bones are imported from another character
    sy = batch_skin_eyes.SkinEyes()
    sy.set_skin()