diff options
Diffstat (limited to 'space_view3d_pie_menus/pie_save_open_menu.py')
| -rw-r--r-- | space_view3d_pie_menus/pie_save_open_menu.py | 217 |
1 files changed, 217 insertions, 0 deletions
diff --git a/space_view3d_pie_menus/pie_save_open_menu.py b/space_view3d_pie_menus/pie_save_open_menu.py new file mode 100644 index 00000000..2b8e4e87 --- /dev/null +++ b/space_view3d_pie_menus/pie_save_open_menu.py @@ -0,0 +1,217 @@ +# ##### BEGIN GPL LICENSE BLOCK ##### +# +# This program is free software; you can redistribute it and/or +# modify it under the terms of the GNU General Public License +# as published by the Free Software Foundation; either version 2 +# of the License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software Foundation, +# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. +# +# ##### END GPL LICENSE BLOCK ##### + +# <pep8 compliant> + +bl_info = { + "name": "Hotkey: 'Ctrl S'", + "description": "Save/Open & File Menus", + "blender": (2, 77, 0), + "location": "All Editors", + "warning": "", + "wiki_url": "", + "category": "Save Open Pie" + } + +import bpy +from bpy.types import ( + Menu, + Operator, + ) +import os + +# Pie Save/Open + + +class PieSaveOpen(Menu): + bl_idname = "pie.saveopen" + bl_label = "Pie Save/Open" + + def draw(self, context): + layout = self.layout + pie = layout.menu_pie() + # 4 - LEFT + pie.operator("wm.read_homefile", text="New", icon='NEW') + # 6 - RIGHT + pie.menu("pie.link", text="Link", icon='LINK_BLEND') + # 2 - BOTTOM + pie.menu("pie.fileio", text="Import/Export", icon='IMPORT') + # 8 - TOP + pie.operator("file.save_incremental", text="Incremental Save", icon='SAVE_COPY') + # 7 - TOP - LEFT + pie.operator("wm.save_mainfile", text="Save", icon='FILE_TICK') + # 9 - TOP - RIGHT + pie.operator("wm.save_as_mainfile", text="Save As...", icon='SAVE_AS') + # 1 - BOTTOM - LEFT + pie.operator("wm.open_mainfile", text="Open file", icon='FILE_FOLDER') + # 3 - BOTTOM - RIGHT + pie.menu("pie.recover", text="Recovery", icon='RECOVER_LAST') + + +class pie_link(Menu): + bl_idname = "pie.link" + bl_label = "Link" + + def draw(self, context): + layout = self.layout + pie = layout.menu_pie() + box = pie.split().column() + row = box.row(align=True) + box.operator("wm.link", text="Link", icon='LINK_BLEND') + box.operator("wm.append", text="Append", icon='APPEND_BLEND') + box.menu("external.data", text="External Data", icon='EXTERNAL_DATA') + + +class pie_recover(Menu): + bl_idname = "pie.recover" + bl_label = "Recovery" + + def draw(self, context): + layout = self.layout + pie = layout.menu_pie() + box = pie.split().column() + row = box.row(align=True) + box.operator("wm.recover_auto_save", text="Recover Auto Save...", icon='RECOVER_AUTO') + box.operator("wm.recover_last_session", text="Recover Last Session", icon='RECOVER_LAST') + box.operator("wm.revert_mainfile", text="Revert", icon='FILE_REFRESH') + + +class pie_fileio(Menu): + bl_idname = "pie.fileio" + bl_label = "Import/Export" + + def draw(self, context): + layout = self.layout + pie = layout.menu_pie() + box = pie.split().column() + row = box.row(align=True) + box.menu("INFO_MT_file_import", icon='IMPORT') + box.separator() + box.menu("INFO_MT_file_export", icon='EXPORT') + + +class ExternalData(Menu): + bl_idname = "external.data" + bl_label = "External Data" + + def draw(self, context): + layout = self.layout + + layout.operator("file.autopack_toggle", text="Automatically Pack Into .blend") + layout.separator() + layout.operator("file.pack_all", text="Pack All Into .blend") + layout.operator("file.unpack_all", text="Unpack All Into Files") + layout.separator() + layout.operator("file.make_paths_relative", text="Make All Paths Relative") + layout.operator("file.make_paths_absolute", text="Make All Paths Absolute") + layout.operator("file.report_missing_files", text="Report Missing Files") + layout.operator("file.find_missing_files", text="Find Missing Files") + + +# Save Incremental + +class FileIncrementalSave(Operator): + bl_idname = "file.save_incremental" + bl_label = "Save Incremental" + bl_description = "Save First!then Incremental, .blend will get _001 extension" + bl_options = {"REGISTER"} + + @classmethod + def poll(cls, context): + return (bpy.data.filepath is not "") + + def execute(self, context): + f_path = bpy.data.filepath + b_name = bpy.path.basename(f_path) + + if b_name and b_name.find("_") != -1: + # except in cases when there is an underscore in the name like my_file.blend + try: + str_nb = b_name.rpartition("_")[-1].rpartition(".blend")[0] + int_nb = int(str(str_nb)) + new_nb = str_nb.replace(str(int_nb), str(int_nb + 1)) + output = f_path.replace(str_nb, new_nb) + + i = 1 + while os.path.isfile(output): + str_nb = b_name.rpartition("_")[-1].rpartition(".blend")[0] + i += 1 + new_nb = str_nb.replace(str(int_nb), str(int_nb + i)) + output = f_path.replace(str_nb, new_nb) + except ValueError: + output = f_path.rpartition(".blend")[0] + "_001" + ".blend" + else: + # no underscore in the name or saving a nameless (.blend) file + output = f_path.rpartition(".blend")[0] + "_001" + ".blend" + + # fix for saving in a directory without privileges + try: + bpy.ops.wm.save_as_mainfile(filepath=output) + except: + self.report({'WARNING'}, "File could not be saved. Check the System Console for errors") + return {'CANCELLED'} + + self.report( + {'INFO'}, "File: {0} - Created at: {1}".format( + output[len(bpy.path.abspath("//")):], + output[:len(bpy.path.abspath("//"))]), + ) + return {'FINISHED'} + + +classes = ( + PieSaveOpen, + ExternalData, + FileIncrementalSave, + pie_fileio, + pie_recover, + pie_link, + ) + +addon_keymaps = [] + + +def register(): + for cls in classes: + bpy.utils.register_class(cls) + wm = bpy.context.window_manager + + if wm.keyconfigs.addon: + # Save/Open/... + km = wm.keyconfigs.addon.keymaps.new(name='Window') + kmi = km.keymap_items.new('wm.call_menu_pie', 'S', 'PRESS', ctrl=True) + kmi.properties.name = "pie.saveopen" +# kmi.active = True + addon_keymaps.append((km, kmi)) + + +def unregister(): + for cls in classes: + bpy.utils.unregister_class(cls) + wm = bpy.context.window_manager + + kc = wm.keyconfigs.addon + if kc: + km = kc.keymaps['Window'] + for kmi in km.keymap_items: + if kmi.idname == 'wm.call_menu_pie': + if kmi.properties.name == "pie.saveopen": + km.keymap_items.remove(kmi) + +if __name__ == "__main__": + register() |