Add photo management scripts
parent
a8b69fec4c
commit
d28a647895
GPG Key ID:
7E6D18B61C8D2F4B
@ -0,0 +1,173 @@
|
||||
-- Fork of the autogroup plugin but only comparing pictures with the same timestamp
|
||||
local dt = require "darktable"
|
||||
local du = require "lib/dtutils"
|
||||
local df = require "lib/dtutils.file"
|
||||
|
||||
du.check_min_api_version("7.0.0", "AutoGroupInstant")
|
||||
|
||||
local MOD = 'autogroupinstant'
|
||||
|
||||
-- return data structure for script_manager
|
||||
|
||||
local script_data = {}
|
||||
|
||||
script_data.destroy = nil -- function to destory the script
|
||||
script_data.destroy_method = nil -- set to hide for libs since we can't destroy them commpletely yet, otherwise leave as nil
|
||||
script_data.restart = nil -- how to restart the (lib) script after it's been hidden - i.e. make it visible again
|
||||
script_data.show = nil -- only required for libs since the destroy_method only hides them
|
||||
|
||||
local gettext = dt.gettext
|
||||
-- Tell gettext where to find the .mo file translating messages for a particular domain
|
||||
gettext.bindtextdomain("AutoGroupInstant",dt.configuration.config_dir.."/lua/locale/")
|
||||
|
||||
local function _(msgid)
|
||||
return gettext.dgettext("AutoGroupInstant", msgid)
|
||||
end
|
||||
|
||||
local Ag = {}
|
||||
Ag.module_installed = false
|
||||
Ag.event_registered = false
|
||||
|
||||
local GUI = {
|
||||
gap = {},
|
||||
selected = {},
|
||||
collection = {}
|
||||
}
|
||||
|
||||
|
||||
local function InRange(test, low, high) --tests if test value is within range of low and high (inclusive)
|
||||
if test >= low and test <= high then
|
||||
return true
|
||||
else
|
||||
return false
|
||||
end
|
||||
end
|
||||
|
||||
local function CompTime(first, second) --compares the timestamps and returns true if first was taken before second
|
||||
first_time = first.exif_datetime_taken
|
||||
if string.match(first_time, '[0-9]') == nil then first_time = '9999:99:99 99:99:99' end
|
||||
first_time = tonumber(string.gsub(first_time, '[^0-9]*',''))
|
||||
second_time = second.exif_datetime_taken
|
||||
if string.match(second_time, '[0-9]') == nil then second_time = '9999:99:99 99:99:99' end
|
||||
second_time = tonumber(string.gsub(second_time, '[^0-9]*',''))
|
||||
return first_time < second_time
|
||||
end
|
||||
|
||||
local function SeperateTime(str) --seperates the timestamp into individual components for used with OS.time operations
|
||||
local cleaned = string.gsub(str, '[^%d]',':')
|
||||
cleaned = string.gsub(cleaned, '::*',':') --YYYY:MM:DD:hh:mm:ss
|
||||
local year = string.sub(cleaned,1,4)
|
||||
local month = string.sub(cleaned,6,7)
|
||||
local day = string.sub(cleaned,9,10)
|
||||
local hour = string.sub(cleaned,12,13)
|
||||
local min = string.sub(cleaned,15,16)
|
||||
local sec = string.sub(cleaned,18,19)
|
||||
return {year = year, month = month, day = day, hour = hour, min = min, sec = sec}
|
||||
end
|
||||
|
||||
local function GetTimeDiff(curr_image, prev_image) --returns the time difference (in sec.) from current image and the previous image
|
||||
local curr_time = SeperateTime(curr_image.exif_datetime_taken)
|
||||
local prev_time = SeperateTime(prev_image.exif_datetime_taken)
|
||||
return os.time(curr_time)-os.time(prev_time)
|
||||
end
|
||||
|
||||
local function main(on_collection)
|
||||
local images = {}
|
||||
if on_collection then
|
||||
local col_images = dt.collection
|
||||
for i,image in ipairs(col_images) do --copy images to a standard table, table.sort barfs on type dt_lua_singleton_image_collection
|
||||
table.insert(images,i,image)
|
||||
end
|
||||
else
|
||||
images = dt.gui.selection()
|
||||
end
|
||||
if #images < 2 then
|
||||
dt.print('please select at least 2 images')
|
||||
return
|
||||
end
|
||||
table.sort(images, function(first, second) return CompTime(first,second) end) --sort images by timestamp
|
||||
|
||||
for i, image in ipairs(images) do
|
||||
if i == 1 then
|
||||
prev_image = image
|
||||
elseif string.match(image.exif_datetime_taken, '[%d]') ~= nil then --make sure current image has a timestamp
|
||||
local curr_image = image
|
||||
|
||||
if GetTimeDiff(curr_image, prev_image) == 0 then
|
||||
images[i]:group_with(images[i-1])
|
||||
|
||||
for _, member in ipairs(images[i]:get_group_members()) do
|
||||
local ext = string.lower(df.get_filetype(member.filename))
|
||||
if ext == "rw2" or ext == "arw" then
|
||||
member:make_group_leader()
|
||||
end
|
||||
end
|
||||
end
|
||||
prev_image = curr_image
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local function install_module()
|
||||
if not Ag.module_installed then
|
||||
dt.print_log("installing module")
|
||||
dt.register_lib(
|
||||
MOD, -- Module name
|
||||
_('Group identical Timestamps'), -- name
|
||||
true, -- expandable
|
||||
true, -- resetable
|
||||
{[dt.gui.views.lighttable] = {"DT_UI_CONTAINER_PANEL_RIGHT_CENTER", 99}}, -- containers
|
||||
dt.new_widget("box"){
|
||||
orientation = "vertical",
|
||||
GUI.selected,
|
||||
GUI.collection
|
||||
}
|
||||
)
|
||||
Ag.module_installed = true
|
||||
dt.print_log("module installed")
|
||||
dt.print_log("styles module visibility is " .. tostring(dt.gui.libs["styles"].visible))
|
||||
end
|
||||
end
|
||||
|
||||
local function destroy()
|
||||
dt.gui.libs[MOD].visible = false
|
||||
end
|
||||
|
||||
local function restart()
|
||||
dt.gui.libs[MOD].visible = true
|
||||
end
|
||||
|
||||
-- GUI --
|
||||
GUI.selected = dt.new_widget("button"){
|
||||
label = _('auto group: selected'),
|
||||
tooltip =_('auto group selected images'),
|
||||
clicked_callback = function() main(false) end
|
||||
}
|
||||
GUI.collection = dt.new_widget("button"){
|
||||
label = _('auto group: collection'),
|
||||
tooltip =_('auto group the entire collection'),
|
||||
clicked_callback = function() main(true) end
|
||||
}
|
||||
|
||||
if dt.gui.current_view().id == "lighttable" then
|
||||
install_module()
|
||||
else
|
||||
if not Ag.event_registered then
|
||||
dt.register_event(
|
||||
"AutoGroupInstant", "view-changed",
|
||||
function(event, old_view, new_view)
|
||||
if new_view.name == "lighttable" and old_view.name == "darkroom" then
|
||||
install_module()
|
||||
end
|
||||
end
|
||||
)
|
||||
Ag.event_registered = true
|
||||
end
|
||||
end
|
||||
|
||||
script_data.destroy = destroy
|
||||
script_data.destroy_method = "hide"
|
||||
script_data.restart = restart
|
||||
script_data.show = restart
|
||||
|
||||
return script_data
|
||||
@ -0,0 +1,4 @@
|
||||
require "tools/script_manager"
|
||||
require "AutoGroupInstant"
|
||||
|
||||
|
||||
@ -0,0 +1,109 @@
|
||||
def main [] {
|
||||
}
|
||||
|
||||
export def `main import` [
|
||||
src: string # source path
|
||||
dst: string # destination path
|
||||
--exif-only # only take the exif date
|
||||
--link # replace the originals with symlinks
|
||||
--tag-origin # tag with origin folder name
|
||||
] {
|
||||
print "Importing"
|
||||
if ("moved.csv" | path exists) == false {
|
||||
"source,destination\n" | save moved.csv
|
||||
}
|
||||
let entries = ( glob $"($src | str trim --right --char '/')/**/*"
|
||||
| each { try { ls -l $in } catch {|e| print_line -e $"Failed to get metadata of ($e)"; []} }
|
||||
| where { ($in | length) > 0 }
|
||||
| each { first }
|
||||
| where type == 'file'
|
||||
| where size > 0B
|
||||
)
|
||||
let total_count = $entries | length
|
||||
print $"Copying ($total_count) entries"
|
||||
|
||||
( $entries
|
||||
| enumerate
|
||||
| par-each {|entry|
|
||||
let file = $entry.item
|
||||
print_line $file.name
|
||||
progress $entry.index $total_count
|
||||
let meta = exiftool -j $file.name | from json | get 0
|
||||
mut date = $file.created?
|
||||
|
||||
if $date == null {
|
||||
$date = $file.modified?
|
||||
}
|
||||
|
||||
let path_parts = $file.name | path parse
|
||||
mut using_exif = false
|
||||
|
||||
if $meta.DateTimeOriginal? != null {
|
||||
try {
|
||||
$date = (parse_exif_timestamp $meta.DateTimeOriginal $meta.OffsetTimeOriginal?)
|
||||
$using_exif = true
|
||||
} catch {|e|
|
||||
print_line -e $"Failed to parse datetime ($meta.DateTimeOriginal) ($e)"
|
||||
}
|
||||
}
|
||||
if $using_exif == false and $exif_only {
|
||||
let folder_rel = ( try {
|
||||
$path_parts.parent | path relative-to $src | into string
|
||||
} catch {
|
||||
$path_parts.parent | path parse | get stem
|
||||
})
|
||||
let folder = $dst | path join "unknown" | path join $folder_rel
|
||||
mkdir $folder
|
||||
|
||||
let file_name = $folder | path join $"($path_parts.stem).($path_parts.extension)"
|
||||
archive-cp $file.name $file_name --tag=$tag_origin
|
||||
|
||||
print_line $"Inaccurate date for file ($file.name)"
|
||||
progress ($entry.index + 1) $total_count
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
let file_name = $"($date | format date '%Y-%m-%d_%H%M%S')_($path_parts.stem).($path_parts.extension)"
|
||||
print_line $"Filename: ($file_name)"
|
||||
progress $entry.index $total_count
|
||||
let folder_name = $dst | path join ($date | format date "%Y/%Y-%m-%B")
|
||||
|
||||
print $"Destination folder: ($folder_name)"
|
||||
mkdir $folder_name
|
||||
archive-cp $file.name ($folder_name | path join $file_name) --link=$link --tag=$tag_origin
|
||||
|
||||
print_line $"Copied ($meta.SourceFile)"
|
||||
progress ($entry.index + 1) $total_count
|
||||
})
|
||||
}
|
||||
|
||||
def parse_exif_timestamp [timestamp: string, offset?: any] {
|
||||
let components = $timestamp | split row " "
|
||||
let date = $components | first
|
||||
let time = $components | last
|
||||
|
||||
$"($date | str replace -a ':' '.') ($time)($offset)" | into datetime
|
||||
}
|
||||
|
||||
def archive-cp [src: string, dst: string, --link, --tag] {
|
||||
rsync -X -U -p -t -g -o -u --info=ALL $src $dst
|
||||
$"($src),($dst)\n" | save -a moved.csv
|
||||
|
||||
if $tag {
|
||||
xattr -w user.xdg.tags ($src | path parse | get parent | path parse | get stem) $dst
|
||||
}
|
||||
if $link {
|
||||
rm $src
|
||||
ln -s $dst $src
|
||||
}
|
||||
}
|
||||
|
||||
def progress [current: number, total: number] {
|
||||
let progress_perc = ($current / $total) * 100
|
||||
print $"($current) / ($total) \(($progress_perc | math round -p 2)%\): (0..($progress_perc / 5 | math round) | each { '#' } | str join '')(ansi -e F)"
|
||||
}
|
||||
|
||||
def print_line [...msg: any, -e] {
|
||||
print --stderr=$e $"(ansi -e 2K)( $msg | each { into string } | str join ' ' )"
|
||||
}
|
||||
Loading…
Reference in New Issue