140 lines
3.8 KiB
Python
140 lines
3.8 KiB
Python
# ##### 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 3 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, see <https://www.gnu.org/licenses/>.
|
|
#
|
|
# ##### END GPL LICENSE BLOCK #####
|
|
|
|
|
|
import logging
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import bpy
|
|
import mathutils
|
|
from replication.constants import DIFF_BINARY, UP
|
|
from replication.protocol import ReplicatedDatablock
|
|
|
|
from .. import utils
|
|
from .dump_anything import Dumper, Loader
|
|
|
|
|
|
def get_filepath(filename):
|
|
"""
|
|
Construct the local filepath
|
|
"""
|
|
return str(Path(
|
|
utils.get_preferences().cache_directory,
|
|
filename
|
|
))
|
|
|
|
|
|
def ensure_unpacked(datablock):
|
|
if datablock.packed_file:
|
|
logging.info(f"Unpacking {datablock.name}")
|
|
|
|
filename = Path(bpy.path.abspath(datablock.filepath)).name
|
|
datablock.filepath = get_filepath(filename)
|
|
|
|
datablock.unpack(method="WRITE_ORIGINAL")
|
|
|
|
|
|
class BlFile(ReplicatedDatablock):
|
|
bl_id = 'file'
|
|
bl_name = "file"
|
|
bl_class = Path
|
|
bl_check_common = False
|
|
bl_icon = 'FILE'
|
|
bl_reload_parent = True
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.instance = kwargs.get('instance', None)
|
|
|
|
if self.instance and not self.instance.exists():
|
|
raise FileNotFoundError(str(self.instance))
|
|
|
|
self.preferences = utils.get_preferences()
|
|
|
|
def resolve(self, construct = True):
|
|
self.instance = Path(get_filepath(self.data['name']))
|
|
|
|
file_exists = self.instance.exists()
|
|
if not file_exists:
|
|
logging.debug("File don't exist, loading it.")
|
|
self._load(self.data, self.instance)
|
|
|
|
return file_exists
|
|
|
|
|
|
def push(self, socket, identity=None, check_data=False):
|
|
super().push(socket, identity=None, check_data=False)
|
|
|
|
if self.preferences.clear_memory_filecache:
|
|
del self.data['file']
|
|
|
|
def dump(self, instance=None):
|
|
"""
|
|
Read the file and return a dict as:
|
|
{
|
|
name : filename
|
|
extension :
|
|
file: file content
|
|
}
|
|
"""
|
|
logging.info(f"Extracting file metadata")
|
|
|
|
data = {
|
|
'name': self.instance.name,
|
|
}
|
|
|
|
logging.info(
|
|
f"Reading {self.instance.name} content: {self.instance.stat().st_size} bytes")
|
|
|
|
try:
|
|
file = open(self.instance, "rb")
|
|
data['file'] = file.read()
|
|
|
|
file.close()
|
|
except IOError:
|
|
logging.warning(f"{self.instance} doesn't exist, skipping")
|
|
else:
|
|
file.close()
|
|
|
|
return data
|
|
|
|
def load(self, data, target):
|
|
"""
|
|
Writing the file
|
|
"""
|
|
|
|
try:
|
|
file = open(target, "wb")
|
|
file.write(data['file'])
|
|
|
|
if self.preferences.clear_memory_filecache:
|
|
del self.data['file']
|
|
except IOError:
|
|
logging.warning(f"{target} doesn't exist, skipping")
|
|
else:
|
|
file.close()
|
|
|
|
def diff(self):
|
|
if self.preferences.clear_memory_filecache:
|
|
return False
|
|
else:
|
|
memory_size = sys.getsizeof(self.data['file'])-33
|
|
disk_size = self.instance.stat().st_size
|
|
return memory_size != disk_size
|