source: extensions/piwigo_import_tree/piwigo_import_tree.pl @ 15730

Revision 15730, 6.0 KB checked in by plg, 7 years ago (diff)

version 0.1 (already published)

Line 
1#!/usr/bin/perl
2
3# usage:
4# perl piwigo_import_tree.pl --directory=/Users/pierrick/piwigo/album1
5
6use strict;
7use warnings;
8
9use File::Find;
10use Data::Dumper;
11use File::Basename;
12use LWP::UserAgent;
13use JSON;
14use Getopt::Long;
15use Encode qw/is_utf8 decode/;
16
17my %opt = ();
18GetOptions(
19    \%opt,
20    qw/
21          base_url=s
22          username=s
23          password=s
24          directory=s
25          parent_album_id=s
26          define=s%
27          quiet
28      /
29);
30
31my $album_dir = $opt{directory};
32$album_dir =~ s{^\./*}{};
33
34our $ua = LWP::UserAgent->new;
35$ua->agent('Mozilla/piwigo_remote.pl 1.25');
36$ua->cookie_jar({});
37
38my %conf;
39my %conf_default = (
40    base_url => 'http://localhost/plg/piwigo/salon',
41    username => 'plg',
42    password => 'plg',
43);
44
45foreach my $conf_key (keys %conf_default) {
46    $conf{$conf_key} = defined $opt{$conf_key} ? $opt{$conf_key} : $conf_default{$conf_key}
47}
48
49$ua->default_headers->authorization_basic(
50    $conf{username},
51    $conf{password}
52);
53
54my $result = undef;
55my $query = undef;
56
57binmode STDOUT, ":encoding(utf-8)";
58
59# Login to Piwigo
60my $form = {
61    method => 'pwg.session.login',
62    username => $conf{username},
63    password => $conf{password},
64};
65
66$result = $ua->post(
67    $conf{base_url}.'/ws.php?format=json',
68    $form
69);
70
71# Fill an "album path to album id" cache
72my %piwigo_albums = ();
73
74my $response = $ua->post(
75    $conf{base_url}.'/ws.php?format=json',
76    {
77                method => 'pwg.categories.getList',
78                recursive => 1,
79                fullname => 1,
80        }
81);
82
83my $albums_aref = from_json($response->content)->{result}->{categories};
84foreach my $album_href (@{$albums_aref}) {
85        $piwigo_albums{ $album_href->{name} } = $album_href->{id};
86}
87# print Dumper(\%piwigo_albums)."\n\n";
88
89if (defined $opt{parent_album_id}) {
90        foreach my $album_path (keys %piwigo_albums) {
91                if ($piwigo_albums{$album_path} == $opt{parent_album_id}) {
92                        $conf{parent_album_id} = $opt{parent_album_id};
93                        $conf{parent_album_path} = $album_path;
94                }
95        }       
96        if (not defined $conf{parent_album_path}) {
97                print "Parent album ".$opt{parent_album_id}." does not exist\n";
98                exit();
99        }
100}
101
102# Initialize a cache with file names of existing photos, for related albums
103my %photos_of_album = ();
104
105# Synchronize local folder with remote Piwigo gallery
106find({wanted => \&add_to_piwigo, no_chdir => 1}, $album_dir);
107
108#---------------------------------------------------------------------
109# Functions
110#---------------------------------------------------------------------
111
112sub fill_photos_of_album {
113        my %params = @_;
114       
115        if (defined $photos_of_album{ $params{album_id} }) {
116                return 1;
117        }
118       
119        my $response = $ua->post(
120        $conf{base_url}.'/ws.php?format=json',
121        {
122            method => 'pwg.categories.getImages',
123            cat_id => $params{album_id},
124        }
125    );
126   
127    foreach my $image_href (@{from_json($response->content)->{result}{images}{_content}}) {
128        $photos_of_album{ $params{album_id} }{ $image_href->{file} } = 1;
129    }
130}
131
132sub photo_exists {
133        my %params = @_;
134
135    fill_photos_of_album(album_id => $params{album_id});
136   
137    if (defined $photos_of_album{ $params{album_id} }{ $params{file} }) {
138        return 1;
139    }
140    else {
141        return 0;
142    }
143}
144
145sub add_album {
146        my %params = @_;
147       
148        # print Dumper(\%params);
149       
150    my $form = {
151        method => 'pwg.categories.add',
152        name => $params{name},
153    };
154   
155    if (defined $params{parent}) {
156        $form->{parent} = $params{parent};
157    }
158
159    my $response = $ua->post(
160        $conf{base_url}.'/ws.php?format=json',
161        $form
162    );
163
164    return from_json($response->content)->{result}{id};
165}
166
167sub add_photo {
168    my %params = @_;
169
170    my $form = {
171        method => 'pwg.images.addSimple',
172        image => [$params{path}],
173        category => $params{album_id},
174    };
175
176    # is there any title defined in a descript.ion file?
177    my $property = undef;
178    my $desc_filepath = dirname($params{path}).'/descript.ion';
179    if (-f $desc_filepath) {
180        my $photo_filename = basename($params{path});
181        open(IN, '<', $desc_filepath);
182        while (my $desc_line = <IN>) {
183            if ($desc_line =~ /^$photo_filename/) {
184                chomp($desc_line);
185                $property = (split /\t/, $desc_line, 2)[1];
186            }
187        }
188        close(IN);
189    }
190
191    if (defined $property and $property ne '') {
192        $form->{name} = $property;
193    }
194
195    my $response = $ua->post(
196        $conf{base_url}.'/ws.php?format=json',
197        $form,
198        'Content_Type' => 'form-data',
199    );
200}
201
202sub add_to_piwigo {
203    # print $File::Find::name."\n";
204    my $path = $File::Find::name;
205    my $parent_dir = dirname($album_dir);
206    if ($parent_dir ne '.') {
207        # print '$parent_dir = '.$parent_dir."\n";
208        $path =~ s{^$parent_dir/}{};
209    }
210    # print $path."\n";
211
212    if (-d) {
213        my $up_dir = '';
214        my $parent_id = undef;
215
216        if (defined $conf{parent_album_path}) {
217            $up_dir = $conf{parent_album_path}.' / ';
218            $parent_id = $conf{parent_album_id};
219        }
220
221        foreach my $dir (split '/', $path) {
222            if (not defined $piwigo_albums{$up_dir.$dir}) {
223                print 'album "'.$up_dir.$dir.'" must be created'."\n";
224                my $id = add_album(name => $dir, parent => $parent_id);
225                $piwigo_albums{$up_dir.$dir} = $id;
226            }
227            $parent_id = $piwigo_albums{$up_dir.$dir};
228            $up_dir.= $dir.' / ';
229        }
230    }
231
232    if (-f and $path =~ /\.(jpe?g|gif|png)$/i) {
233        my $album_key = join(' / ', split('/', dirname($path)));
234
235        if (defined $conf{parent_album_path}) {
236            $album_key = $conf{parent_album_path}.' / '.$album_key;
237        }
238
239        my $album_id = $piwigo_albums{$album_key};
240
241        if (photo_exists(album_id => $album_id, file => basename($File::Find::name))) {
242            if (not $opt{quiet}) {
243                print $File::Find::name.' already exists in Piwigo, skipped'."\n";
244            }
245            return 1;
246        }
247
248        print $File::Find::name.' must be uploaded in "'.$album_key.'" (id='.$album_id.')'."\n";
249        add_photo(path => $File::Find::name, album_id => $album_id);
250        # print 'dirname = '.dirname($path)."\n\n";
251    }
252}
Note: See TracBrowser for help on using the repository browser.