mirror of
https://github.com/gryf/pygtktalog.git
synced 2025-12-17 11:30:19 +01:00
gryf
33a8f99d48
Till now there was an idea to expose full path to image file and thumbnail. It's quite impossible in certain situations, where path to the media files is not known. Therefore 'imagepath' was removed due to redundancy with filename stored in db, and 'thumpath', which now is a simple calculation of the expected thumbnail relative filepath.
298 lines
9.3 KiB
Python
298 lines
9.3 KiB
Python
"""
|
|
Project: pyGTKtalog
|
|
Description: Definition of DB objects classes. Using SQLAlchemy.
|
|
Type: core
|
|
Author: Roman 'gryf' Dobosz, gryf73@gmail.com
|
|
Created: 2009-08-07
|
|
"""
|
|
import os
|
|
import shutil
|
|
|
|
from sqlalchemy import Column, Table, Integer, Text
|
|
from sqlalchemy import DateTime, ForeignKey, Sequence
|
|
from sqlalchemy.orm import relation, backref
|
|
|
|
from pygtktalog.dbcommon import Base
|
|
from pygtktalog.thumbnail import ThumbCreator
|
|
from pygtktalog.logger import get_logger
|
|
from pygtktalog.misc import mk_paths
|
|
|
|
|
|
LOG = get_logger(__name__)
|
|
|
|
tags_files = Table("tags_files", Base.metadata,
|
|
Column("file_id", Integer, ForeignKey("files.id")),
|
|
Column("tag_id", Integer, ForeignKey("tags.id")))
|
|
|
|
TYPE = {'root': 0, 'dir': 1, 'file': 2, 'link': 3}
|
|
|
|
|
|
class File(Base):
|
|
"""
|
|
File mapping. Instances of this object can reference other File object
|
|
which make the structure to be tree-like
|
|
"""
|
|
__tablename__ = "files"
|
|
id = Column(Integer, Sequence("file_id_seq"), primary_key=True)
|
|
parent_id = Column(Integer, ForeignKey("files.id"), index=True)
|
|
filename = Column(Text)
|
|
filepath = Column(Text)
|
|
date = Column(DateTime)
|
|
size = Column(Integer)
|
|
type = Column(Integer, index=True)
|
|
source = Column(Integer)
|
|
note = Column(Text)
|
|
description = Column(Text)
|
|
# checksum = Column(Text)
|
|
|
|
children = relation('File',
|
|
backref=backref('parent', remote_side="File.id"),
|
|
order_by=[type, filename])
|
|
tags = relation("Tag", secondary=tags_files, order_by="Tag.tag")
|
|
thumbnail = relation("Thumbnail", backref="file")
|
|
images = relation("Image", backref="file", order_by="Image.filename")
|
|
|
|
def __init__(self, filename=None, path=None, date=None, size=None,
|
|
ftype=None, src=None):
|
|
"""Create file object with empty defaults"""
|
|
self.filename = filename
|
|
self.filepath = path
|
|
self.date = date
|
|
self.size = size
|
|
self.type = ftype
|
|
self.source = src
|
|
|
|
def __repr__(self):
|
|
return "<File('%s', %s)>" % (self.filename.encode('utf-8'),
|
|
str(self.id))
|
|
|
|
def get_all_children(self):
|
|
"""
|
|
Return list of all node direct and indirect children
|
|
"""
|
|
def _recursive(node):
|
|
children = []
|
|
if node.children:
|
|
for child in node.children:
|
|
children += _recursive(child)
|
|
if node != self:
|
|
children.append(node)
|
|
|
|
return children
|
|
|
|
if self.children:
|
|
return _recursive(self)
|
|
else:
|
|
return []
|
|
|
|
|
|
class Group(Base):
|
|
"""TODO: what is this class for?"""
|
|
__tablename__ = "groups"
|
|
id = Column(Integer, Sequence("group_id_seq"), primary_key=True)
|
|
name = Column(Text)
|
|
color = Column(Text)
|
|
|
|
def __init__(self, name=None, color=None):
|
|
self.name = name
|
|
self.color = color
|
|
|
|
def __repr__(self):
|
|
return "<Group('%s', %s)>" % (str(self.name), str(self.id))
|
|
|
|
|
|
class Tag(Base):
|
|
"""Tag mapping"""
|
|
__tablename__ = "tags"
|
|
id = Column(Integer, Sequence("tags_id_seq"), primary_key=True)
|
|
group_id = Column(Integer, ForeignKey("groups.id"), index=True)
|
|
tag = Column(Text)
|
|
group = relation('Group', backref=backref('tags', remote_side="Group.id"))
|
|
|
|
files = relation("File", secondary=tags_files)
|
|
|
|
def __init__(self, tag=None, group=None):
|
|
self.tag = tag
|
|
self.group = group
|
|
|
|
def __repr__(self):
|
|
return "<Tag('%s', %s)>" % (str(self.tag), str(self.id))
|
|
|
|
|
|
class Thumbnail(Base):
|
|
"""Thumbnail for the file"""
|
|
__tablename__ = "thumbnails"
|
|
id = Column(Integer, Sequence("thumbnail_id_seq"), primary_key=True)
|
|
file_id = Column(Integer, ForeignKey("files.id"), index=True)
|
|
filename = Column(Text)
|
|
|
|
def __init__(self, filename=None, img_path=None, file_obj=None):
|
|
self.filename = filename
|
|
self.file = file_obj
|
|
self.img_path = img_path
|
|
if filename and file_obj and img_path:
|
|
self.save(self.filename, img_path)
|
|
|
|
def save(self, fname, img_path):
|
|
"""
|
|
Create file related thumbnail, add it to the file object.
|
|
"""
|
|
new_name = mk_paths(fname, img_path)
|
|
ext = os.path.splitext(self.filename)[1]
|
|
if ext:
|
|
new_name.append("".join([new_name.pop(), ext]))
|
|
|
|
thumb = ThumbCreator(self.filename).generate()
|
|
name, ext = os.path.splitext(new_name.pop())
|
|
new_name.append("".join([name, "_t", ext]))
|
|
self.filename = os.path.sep.join(new_name)
|
|
if not os.path.exists(os.path.join(img_path, *new_name)):
|
|
shutil.move(thumb, os.path.join(img_path, *new_name))
|
|
else:
|
|
LOG.info("Thumbnail already exists (%s: %s)" % \
|
|
(fname, "/".join(new_name)))
|
|
os.unlink(thumb)
|
|
|
|
def __repr__(self):
|
|
return "<Thumbnail('%s', %s)>" % (str(self.filename), str(self.id))
|
|
|
|
|
|
class Image(Base):
|
|
"""Images and their thumbnails"""
|
|
__tablename__ = "images"
|
|
id = Column(Integer, Sequence("images_id_seq"), primary_key=True)
|
|
file_id = Column(Integer, ForeignKey("files.id"), index=True)
|
|
filename = Column(Text)
|
|
|
|
def __init__(self, filename=None, img_path=None, file_obj=None, move=True):
|
|
self.filename = None
|
|
self.file = file_obj
|
|
self.img_path = img_path
|
|
if filename and img_path:
|
|
self.filename = filename
|
|
self.save(filename, img_path, move)
|
|
|
|
def save(self, fname, img_path, move=True):
|
|
"""
|
|
Save and create coressponding thumbnail (note: it differs from file
|
|
related thumbnail!)
|
|
"""
|
|
new_name = mk_paths(fname, img_path)
|
|
ext = os.path.splitext(self.filename)[1]
|
|
|
|
if ext:
|
|
new_name.append("".join([new_name.pop(), ext]))
|
|
|
|
if not os.path.exists(os.path.join(img_path, *new_name)):
|
|
if move:
|
|
shutil.move(self.filename, os.path.join(img_path, *new_name))
|
|
else:
|
|
shutil.copy(self.filename, os.path.join(img_path, *new_name))
|
|
else:
|
|
LOG.warning("Image with same CRC already exists "
|
|
"('%s', '%s')" % (self.filename, "/".join(new_name)))
|
|
|
|
self.filename = os.path.sep.join(new_name)
|
|
|
|
name, ext = os.path.splitext(new_name.pop())
|
|
new_name.append("".join([name, "_t", ext]))
|
|
|
|
if not os.path.exists(os.path.join(img_path, *new_name)):
|
|
thumb = ThumbCreator(os.path.join(img_path, self.filename))
|
|
shutil.move(thumb.generate(), os.path.join(img_path, *new_name))
|
|
else:
|
|
LOG.info("Thumbnail already generated %s" % "/".join(new_name))
|
|
|
|
|
|
def get_copy(self):
|
|
"""
|
|
Create the very same object as self with exception of id field
|
|
"""
|
|
img = Image()
|
|
img.filename = self.filename
|
|
return img
|
|
|
|
@property
|
|
def thumbnail(self):
|
|
"""
|
|
Return path to thumbnail for this image
|
|
"""
|
|
path, fname = os.path.split(self.filename)
|
|
base, ext = os.path.splitext(fname)
|
|
return os.path.join(path, base + "_t" + ext)
|
|
|
|
def __repr__(self):
|
|
return "<Image('%s', %s)>" % (str(self.filename), str(self.id))
|
|
|
|
|
|
class Exif(Base):
|
|
"""Selected EXIF information"""
|
|
__tablename__ = "exif"
|
|
id = Column(Integer, Sequence("exif_id_seq"), primary_key=True)
|
|
file_id = Column(Integer, ForeignKey("files.id"), index=True)
|
|
camera = Column(Text)
|
|
date = Column(Text)
|
|
aperture = Column(Text)
|
|
exposure_program = Column(Text)
|
|
exposure_bias = Column(Text)
|
|
iso = Column(Text)
|
|
focal_length = Column(Text)
|
|
subject_distance = Column(Text)
|
|
metering_mode = Column(Text)
|
|
flash = Column(Text)
|
|
light_source = Column(Text)
|
|
resolution = Column(Text)
|
|
orientation = Column(Text)
|
|
|
|
def __init__(self):
|
|
self.camera = None
|
|
self.date = None
|
|
self.aperture = None
|
|
self.exposure_program = None
|
|
self.exposure_bias = None
|
|
self.iso = None
|
|
self.focal_length = None
|
|
self.subject_distance = None
|
|
self.metering_mode = None
|
|
self.flash = None
|
|
self.light_source = None
|
|
self.resolution = None
|
|
self.orientation = None
|
|
|
|
def __repr__(self):
|
|
return "<Exif('%s', %s)>" % (str(self.date), str(self.id))
|
|
|
|
|
|
class Gthumb(Base):
|
|
"""Gthumb information"""
|
|
__tablename__ = "gthumb"
|
|
id = Column(Integer, Sequence("gthumb_id_seq"), primary_key=True)
|
|
file_id = Column(Integer, ForeignKey("files.id"), index=True)
|
|
note = Column(Text)
|
|
place = Column(Text)
|
|
date = Column(DateTime)
|
|
|
|
def __init__(self, note=None, place=None, date=None):
|
|
self.note = note
|
|
self.place = place
|
|
self.date = date
|
|
|
|
def __repr__(self):
|
|
return "<Gthumb('%s', '%s', %s)>" % (str(self.date), str(self.place),
|
|
str(self.id))
|
|
|
|
|
|
class Config(Base):
|
|
"""Per-database configuration"""
|
|
__tablename__ = "config"
|
|
id = Column(Integer, Sequence("config_id_seq"), primary_key=True)
|
|
key = Column(Text)
|
|
value = Column(Text)
|
|
|
|
def __init__(self, key=None, value=None):
|
|
self.key = key
|
|
self.value = value
|
|
|
|
def __repr__(self):
|
|
return "<Config('%s', '%s')>" % (str(self.key), str(self.value))
|