Python Store.load方法代码示例

# 需要导入模块: from store import Store [as 别名]
# 或者: from store.Store import load [as 别名]
def go_store():
    store = Store()

    store.remove(None)
    store.add(323, 2, 10, 88, 78, 415, 89, 189)
    store.remove(78)

    store.save()
    store.load()

    print store.grep('\d*[02468]$')
    print store.find(89, -1, 415)

    for item in store.items():
        print item

# 需要导入模块: from store import Store [as 别名]
# 或者: from store.Store import load [as 别名]
def init():
    if intconf('use_hdfs_tree') != 0:
        shell_cmd("mv %s %s.bak" % (conf('tree_pb'), conf('tree_pb')))
        shell_cmd("hadoop fs -get %s/%s/%s %s" % (conf('upload_url'), conf('data_dir'), conf('tree_filename'), conf('tree_pb')))

    _LIB_NAME = 'libselector.so'
    ctypes.CDLL(_LIB_NAME, ctypes.RTLD_GLOBAL)

    from store import Store
    from dist_tree import DistTree
    s = Store(conf('tree_store_config'))
    s.load(conf('tree_pb'))
    tree = DistTree()
    tree.set_store(s.get_handle())
    tree.load()

# 需要导入模块: from store import Store [as 别名]
# 或者: from store.Store import load [as 别名]
class GUI(object):
    def __init__(self, filename):
        self.filename = filename
        self.store = Store()
        self.root = tkinter.Tk()
        self.passphrase = None
        self.selected = None
        self.value_changing = False

        left_frame = tkinter.Frame(self.root)
        left_frame.pack(side=tkinter.LEFT, fill=tkinter.Y)
        left_frame.rowconfigure(0, weight=1)

        scroll = tkinter.Scrollbar(left_frame)
        self.keys = tkinter.Listbox(left_frame, exportselection=0)

        scroll.grid(row=0, column=1, sticky=tkinter.N+tkinter.S)
        self.keys.grid(row=0, column=0, sticky=tkinter.N+tkinter.S)

        scroll.config(command=self.keys.yview)
        self.keys.config(yscrollcommand=scroll.set)

        self.keys.bind("<ButtonRelease>", self.on_key_selected)
        self.keys.bind("<KeyRelease>", self.on_key_selected)

        add = tkinter.Button(
            left_frame,
            text="Add",
            command=self.on_add_entry,
        )
        add.grid(row=1, columnspan=2, sticky=tkinter.E+tkinter.W)

        self.value_box = tkinter.Text(self.root)
        self.value_box.pack(side=tkinter.LEFT, fill=tkinter.BOTH)
        self.value_box.bind('<<Modified>>', self.on_value_changed)

        self.root.bind('<Control-s>', self.on_save)

    def run(self):
        if os.path.exists(self.filename):
            dialog = PassphraseDialog(self.root, title='Passphrase')
            if not dialog.run():
                return
            self.passphrase = dialog.passphrase
            self.store.load(self.passphrase, self.filename)
            self.show_keys()
        else:
            dialog = PassphraseDialog(self.root, title='Passphrase', confirm=True)
            if not dialog.run():
                return
            self.passphrase = dialog.passphrase
            self.store.create(self.passphrase)

        self.update_modified()
        self.root.mainloop()

    def on_key_selected(self, *_args):
        selected = self.keys.curselection()
        if not selected:
            self.selected = None
            return
        idx = int(selected[0])
        self.selected = self.keys.get(idx)
        value = self.store.entries[self.selected]
        self.value_changing = True
        self.value_box.delete(1.0, tkinter.END)
        self.value_box.insert(1.0, value)
        self.value_changing = False

    def on_add_entry(self, *_args):
        dialog = Dialog(self.root, title='Add', label='name')
        if not dialog.run():
            return
        key = dialog.value
        self.store.entries[key] = ''
        self.store.modified = True
        self.update_modified()
        keys = self.show_keys()
        self.keys.select_clear(0, tkinter.END)
        self.keys.select_set(keys.index(key))
        self.on_key_selected()

    def on_value_changed(self, *_args):
        # guard against getting called during key selection changes
        # and also when the modified flag is reset to False.
        if self.value_changing:
            self.root.call(self.value_box, 'edit', 'modified', 0)
            return
        self.value_changing = True
        self.root.call(self.value_box, 'edit', 'modified', 0)
        self.value_changing = False

        if self.selected:
            value = self.value_box.get(1.0, tkinter.END)[:-1]
            if self.store.entries[self.selected] != value:
                self.store.entries[self.selected] = value
                self.store.modified = True
                self.update_modified()

    def on_save(self, *_args):
#.........这里部分代码省略.........

# 需要导入模块: from store import Store [as 别名]
# 或者: from store.Store import load [as 别名]
class CLI(object):
    def __init__(self, filename):
        self.filename = filename
        self.store = Store()

        if os.path.exists(filename):
            self.passphrase = self.get_passphrase()
            self.store.load(self.passphrase, filename)
        else:
            self.passphrase = self.get_passphrase(confirm=True)
            self.store.create(self.passphrase)

        self.commands = {
            'ls' : (self.cmd_list, 'l'),
            'cat' : (self.cmd_cat, 'c:'),
            'set' : (self.cmd_set, 's:'),
            'rm' : (self.cmd_remove, 'r:'),
            'save' : (self.cmd_save, ' '),
            'change_passphrase': (self.cmd_change_passphrase, ' '),
        }


    def cmd_list(self):
        print '\n'.join(sorted(self.store.entries.iterkeys()))

    def cmd_cat(self, label):
        entry = self.store.entries.get(label)
        if entry is not None:
            print entry
        else:
            print 'ERROR: label not found'

    def cmd_set(self, label):
        value = ''
        while True:
            line = sys.stdin.readline()
            if line[0] == '\r' or line[0] == '\n':
                break
            value += line
        self.store.entries[label] = value.strip()
        self.store.modified = True

    def cmd_remove(self, label):
        try:
            del self.store.entries[label]
            self.store.modified = True
        except KeyError:
            print 'ERROR: label not found'

    def cmd_save(self):
        self.store.save(self.filename)

    def cmd_change_passphrase(self):
        current = self.get_passphrase(prompt="Current passphrase: ")
        if current != self.passphrase:
            print 'Incorrect, passphrase not changed.'
            return
        new_passphrase = self.get_passphrase(confirm=True, allow_blank=True)
        if not new_passphrase:
            print 'Passphrase not changed.'
        else:
            self.passphrase = new_passphrase
            self.store.change_passphrase(new_passphrase)

    def completer(self, text, state):
        parts = readline.get_line_buffer().split(' ')
        idx = len(parts)

        if idx == 1:
            matches = 0
            for command, attrs in self.commands.items():
                if command.startswith(text):
                    matches += 1
                    if matches > state:
                        if attrs[-1][-1] == ':':
                            command += ' '
                        return command
        elif idx == 2:
            matches = 0
            for label in self.store.entries.iterkeys():
                if label.startswith(text):
                    matches += 1
                    if matches > state:
                        return label

    def get_passphrase(self, prompt=None, confirm=False, allow_blank=False):
        while True:
            passphrase = getpass(prompt or "Passphrase: ")
            if not passphrase:
                if allow_blank:
                    return
                continue

            if confirm:
                again = getpass("Confirm passphrase: ")
                if passphrase != again:
                    print 'Passphrases did not match, please try again.'
                    continue

            break
#.........这里部分代码省略.........