Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F80256359
websubmit_icon_creator.py
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Fri, Aug 30, 18:04
Size
35 KB
Mime Type
text/x-python
Expires
Sun, Sep 1, 18:04 (2 d)
Engine
blob
Format
Raw Data
Handle
20256286
Attached To
R3600 invenio-infoscience
websubmit_icon_creator.py
View Options
# -*- coding: utf-8 -*-
##
## This file is part of CDS Invenio.
## Copyright (C) 2002, 2003, 2004, 2005, 2006, 2007 CERN.
##
## CDS Invenio 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.
##
## CDS Invenio 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 CDS Invenio; if not, write to the Free Software Foundation, Inc.,
## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
"""This is websubmit_icon_creator.py
This tool is used to create an icon of a picture file.
+ Python API:
Please see create_icon().
+ CLI API:
## $ python ~invenio/lib/python/invenio/websubmit_icon_creator.py \\
## --icon-scale=200 \\
## --icon-name=test-icon \\
## --icon-file-format=jpg \\
## test-image.jpg
## $ python ~invenio/lib/python/invenio/websubmit_icon_creator.py \\
## --icon-scale=200 \\
## --icon-name=test-icon2 \\
## --icon-file-format=gif \\
## --multipage-icon \\
## --multipage-icon-delay=50 \\
## test-image2.pdf
"""
__revision__
=
"$Id$"
import
os.path
,
sys
,
getopt
,
shutil
,
tempfile
,
re
from
invenio.config
import
\
CFG_TMPDIR
,
\
CFG_PATH_DISTILLER
,
\
CFG_PATH_PDFTK
,
\
CFG_PATH_CONVERT
from
invenio.shellutils
import
escape_shell_arg
from
invenio.websubmit_config
import
InvenioWebSubmitIconCreatorError
CFG_ALLOWED_FILE_EXTENSIONS
=
[
"pdf"
,
"gif"
,
"jpg"
,
\
"jpeg"
,
"ps"
,
"png"
,
"bmp"
,
\
"eps"
,
"epsi"
,
"epsf"
]
## ***** Functions related to the icon creation process: *****
# Accepted format for the ImageMagick 'scale' parameter:
re_imagemagic_scale_parameter_format
=
re
.
compile
(
r'x?\d+(x\d*)?(^|!|>|<|@|%)?$'
)
def
create_working_directory
():
"""Create a "working directory" in which the files related to the icon-
creation process can be stored, and return the full path to it.
The working directory will be created in ~invenio/var/tmp.
If it cannot be created there, an exception
(InvenioWebSubmitIconCreatorError) will be raised.
The working directory will have the prefix
"websubmit_icon_creator_", and could be given a name something like:
- websubmit_icon_creator_Tzs3St
@return: (string) - the full path to the working directory.
@Exceptions raised: InvenioWebSubmitIconCreatorError.
"""
## Create the temporary directory in which to place the files related to
## icon creation in ~invenio/var/tmp:
path_workingdir
=
None
try
:
path_workingdir
=
tempfile
.
mkdtemp
(
prefix
=
"websubmit_icon_creator_"
,
\
dir
=
"
%s
"
%
CFG_TMPDIR
)
except
OSError
,
err
:
## Unable to create the temporary directory in ~invenio/var/tmp
msg
=
"Error: Unable to create a temporary working directory in "
\
"which to carry out the icon creation process. An attempt was "
\
"made to create the directory in [
%s
]; the error encountered "
\
"was <
%s
>. Icon creation has failed."
%
(
CFG_TMPDIR
,
str
(
err
))
raise
InvenioWebSubmitIconCreatorError
(
msg
)
## return the path to the working-directory:
return
path_workingdir
def
copy_file_to_directory
(
source_file
,
destination_dir
):
"""Attempt to copy an ordinary file from one location to a destination
directory, returning the name of the copied file if successful.
@param source_file: (string) - the name of the file to be copied
to the destination directory.
@param destination_dir: (string) - the path of the directory into
which the source file is to be copied.
@return: (string) - the name of the source file after it has been
copied to the destination directory (i.e. no leading path information.)
@Exceptions raised: (IOError) - upon failure to successfully copy the
source file to the destination directory.
"""
## Divide the input filename into path and basename:
(
dummy
,
name_source_file
)
=
os
.
path
.
split
(
source_file
)
if
name_source_file
==
""
:
## The source file is just a path - not a valid filename.
msg
=
"""Error: the name of the file to be copied was invalid."""
raise
IOError
(
msg
)
## Test to see whether source file is a real file and is readable:
if
os
.
access
(
"
%s
"
%
source_file
,
os
.
R_OK
):
## File is readable. Copy it locally to the destination directory:
try
:
shutil
.
copyfile
(
"
%s
"
%
source_file
,
\
"
%s
/
%s
"
%
(
destination_dir
,
name_source_file
))
except
IOError
:
## Unable to copy the source file to the destination directory.
msg
=
"""Error: Unable to copy source file [%s] to """
\
"""the destination directory [%s]."""
\
%
(
source_file
,
destination_dir
)
raise
IOError
(
msg
)
else
:
## Unable to read the source file.
msg
=
"""Error: Unable to copy source file [%s] to """
\
"""destination directory [%s]. (File not readable.)"""
\
%
(
source_file
,
destination_dir
)
raise
IOError
(
msg
)
## Now that the source file has been successfully copied to the destination
## directory, return its base name:
return
name_source_file
def
build_icon
(
path_workingdir
,
source_filename
,
source_filetype
,
icon_name
,
icon_filetype
,
multipage_icon
,
multipage_icon_delay
,
icon_scale
):
"""Whereas create_icon acts as the API for icon creation and therefore
deals with argument washing, temporary working directory creation,
etc, the build_icon function takes care of the actual creation of the
icon file itself by calling various shell tools.
To accomplish this, it relies upon the following parameters:
@param path_workingdir: (string) - the path to the working directory
in which all files related to the icon creation are stored.
@param source_filename: (string) - the filename of the original image
file.
@param source_filetype: (string) - the file type of the original image
file.
@param icon_name: (string) - the name that is to be given to the icon.
@param icon_filetype: (string) - the file type of the icon that is
to be created.
@param multipage_icon: (boolean) - a flag indicating whether or not
an icon with multiple pages (i.e. an animated gif icon) should be
created.
@param multipage_icon_delay: (integer) - the delay to be used between
frame changing for an icon with multiple pages (i.e. an animated gif.)
@param icon_scale: (integer) - the scaling information for the created
icon.
@return: (string) - the name of the created icon file (which will have
been created in the working directory "path_workingdir".)
@Exceptions raised: (InvenioWebSubmitIconCreatorError) - raised when
the icon creation process fails.
"""
##
## If the source file is a PS, convert it into a PDF:
if
source_filetype
==
"ps"
:
## Convert the subject file from PostScript to PDF:
if
source_filename
[
-
3
:]
.
lower
()
==
".ps"
:
## The name of the file to be stamped has a PostScript extension.
## Strip it and give the name of the PDF file to be created a
## PDF extension:
created_pdfname
=
"
%s
.pdf"
%
source_filename
[:
-
3
]
elif
len
(
source_filename
.
split
(
"."
))
>
1
:
## The file name has an extension - strip it and add a PDF
## extension:
raw_name
=
source_filename
[:
source_filename
.
rfind
(
"."
)]
if
raw_name
!=
""
:
created_pdfname
=
"
%s
.pdf"
%
raw_name
else
:
## It would appear that the file had no extension and that its
## name started with a period. Just use the original name with
## a .pdf suffix:
created_pdfname
=
"
%s
.pdf"
%
source_filename
else
:
## No extension - use the original name with a .pdf suffix:
created_pdfname
=
"
%s
.pdf"
%
source_filename
## Build the distilling command:
cmd_distill
=
"""%(distiller)s %(ps-file-path)s """
\
"""%(pdf-file-path)s 2>/dev/null"""
%
\
{
'distiller'
:
CFG_PATH_DISTILLER
,
'ps-file-path'
:
escape_shell_arg
(
"
%s
/
%s
"
%
\
(
path_workingdir
,
\
source_filename
)),
'pdf-file-path'
:
escape_shell_arg
(
"
%s
/
%s
"
%
\
(
path_workingdir
,
\
created_pdfname
)),
}
## Distill the PS into a PDF:
errcode_distill
=
os
.
system
(
cmd_distill
)
## Test to see whether the PS was distilled into a PDF without error:
if
errcode_distill
or
\
not
os
.
access
(
"
%s
/
%s
"
%
(
path_workingdir
,
created_pdfname
),
os
.
F_OK
):
## The PDF file was not correctly created in the working directory.
## Unable to continue.
msg
=
"Error: Unable to correctly convert PostScript file [
%s
] to"
\
" PDF. Cannot create icon."
%
source_filename
raise
InvenioWebSubmitIconCreatorError
(
msg
)
## Now assign the name of the created PDF file to subject_file:
source_filename
=
created_pdfname
##
## Treat the name of the icon:
if
icon_name
in
(
None
,
""
):
## Since no name has been provided for the icon, give it the same name
## as the source file, but with the prefix "icon-":
icon_name
=
"icon-
%s
"
%
source_filename
## Now if the icon name has an extension, strip it and add that of the
## icon file type:
if
len
(
icon_name
.
split
(
"."
))
>
1
:
## The icon file name has an extension - strip it and add the icon
## file type extension:
raw_name
=
icon_name
[:
icon_name
.
rfind
(
"."
)]
if
raw_name
!=
""
:
icon_name
=
"
%s
.
%s
"
%
(
raw_name
,
icon_filetype
)
else
:
## It would appear that the file had no extension and that its
## name started with a period. Just use the original name with
## the icon file type's suffix:
icon_name
=
"
%s
.
%s
"
%
(
icon_name
,
icon_filetype
)
else
:
## The icon name had no extension. Use the original name with the
## icon file type's suffix:
icon_name
=
"
%s
.
%s
"
%
(
icon_name
,
icon_filetype
)
##
## If the source file type is PS or PDF, it may be necessary to separate
## the first page from the rest of the document and keep it for use as
## the icon. Do this if necessary:
if
source_filetype
in
(
"ps"
,
"pdf"
)
and
\
(
icon_filetype
!=
"gif"
or
not
multipage_icon
):
## Either (a) the icon type isn't GIF (in which case it cannot
## be animated and must therefore be created _only_ from the
## document's first page; or (b) the icon type is GIF, but the
## icon is to be created from the first page of the document only.
## The first page of the PDF document must be separated and is to
## be used for icon creation:
source_file_first_page
=
"p1-
%s
"
%
source_filename
## Perform the separation:
cmd_get_first_page
=
\
"
%(pdftk)s
A=%(source-file-path)s "
\
"cat A1 output %(first-page-path)s "
\
"2>/dev/null"
\
%
{
'pdftk'
:
CFG_PATH_PDFTK
,
'source-file-path'
:
escape_shell_arg
(
"
%s
/
%s
"
%
\
(
path_workingdir
,
source_filename
)),
'first-page-path'
:
escape_shell_arg
(
"
%s
/
%s
"
%
\
(
path_workingdir
,
\
source_file_first_page
)),
}
errcode_get_first_page
=
os
.
system
(
cmd_get_first_page
)
## Check that the separation was successful:
if
errcode_get_first_page
or
\
not
os
.
access
(
"
%s
/
%s
"
%
(
path_workingdir
,
\
source_file_first_page
),
os
.
F_OK
):
## Separation was unsuccessful.
msg
=
"Error: Unable to create an icon for file [
%s
/
%s
] - it "
\
"wasn't possible to separate the first page from the "
\
"rest of the document (error code [
%s
].)"
\
%
(
path_workingdir
,
source_filename
,
errcode_get_first_page
)
raise
InvenioWebSubmitIconCreatorError
(
msg
)
else
:
## Successfully extracted the first page. Treat it as the source
## file for icon creation from now on:
source_filename
=
source_file_first_page
##
## Create the icon:
## If a delay is necessary for an animated gif icon, create the
## delay string:
delay_info
=
""
if
source_filetype
in
(
"ps"
,
"pdf"
)
and
\
icon_filetype
==
"gif"
and
multipage_icon
:
## Include delay information:
delay_info
=
"-delay
%s
"
%
escape_shell_arg
(
str
(
multipage_icon_delay
))
## Command for icon creation:
cmd_create_icon
=
"
%(convert)s
-scale
%(scale)s
%(delay)s
"
\
"%(source-file-path)s %(icon-file-path)s 2>/dev/null"
\
%
{
'convert'
:
CFG_PATH_CONVERT
,
'scale'
:
\
escape_shell_arg
(
icon_scale
),
'delay'
:
delay_info
,
'source-file-path'
:
\
escape_shell_arg
(
"
%s
/
%s
"
\
%
(
path_workingdir
,
\
source_filename
)),
'icon-file-path'
:
\
escape_shell_arg
(
"
%s
/
%s
"
\
%
(
path_workingdir
,
\
icon_name
)),
}
errcode_create_icon
=
os
.
system
(
cmd_create_icon
)
## Check that the icon creation was successful:
if
errcode_create_icon
or
\
not
os
.
access
(
"
%s
/
%s
"
%
(
path_workingdir
,
icon_name
),
os
.
F_OK
):
## Icon creation was unsuccessful.
msg
=
"Error: Unable to create an icon for file [
%s
/
%s
] (error "
\
"code [
%s
].)"
\
%
(
path_workingdir
,
source_filename
,
errcode_create_icon
)
raise
InvenioWebSubmitIconCreatorError
(
msg
)
##
## The icon was successfully created. Return its name:
return
icon_name
def
create_icon
(
options
):
"""The driver for the icon creation process. This is effectively the
function that is responsible for coordinating the icon creation.
It is the API for the creation of an icon.
@param options: (dictionary) - a dictionary of options that are required
by the function in order to carry out the icon-creation process.
The dictionary must have the following structure:
+ input-file: (string) - the path to the input file (i.e. that
which is to be stamped;
+ icon-name: (string) - the name of the icon that is to be created
by the program. This is optional - if not provided,
a default name will be applied to the icon file instead;
+ multipage-icon: (boolean) - used only when the original file
is a PDF or PS file. If False, the created icon will feature ONLY
the first page of the PDF. If True, ALL pages of the PDF will
be included in the created icon. Note: If the icon type is not
gif, this flag will be forced as False.
+ multipage-icon-delay: (integer) - used only when the original
file is a PDF or PS AND use-first-page-only is False AND
the icon type is gif.
This allows the user to specify the delay between "pages"
of a multi-page (animated) icon.
+ icon-scale: ('geometry') - the scaling information to be used for the
creation of the new icon. Type 'geometry' as defined in ImageMagick.
(eg. 320 or 320x240 or 100> or 5%)
+ icon-file-format: (string) - the file format of the icon that is
to be created. Legal values are:
* pdf
* gif
* jpg
* jpeg
* ps
* png
* bmp
+ verbosity: (integer) - the verbosity level under which the program
is to run;
So, an example of the returned dictionary could be something like:
{ 'input-file' : "demo-picture-file.jpg",
'icon-name' : "icon-demo-picture-file",
'icon-file-format' : "gif",
'multipage-icon' : True,
'multipage-icon-delay' : 100,
'icon-scale' : 180,
'verbosity' : 0,
}
@return: (tuple) - consisting of two strings:
1. the path to the working directory in which all files related to
icon creation are stored;
2. The name of the "icon" file;
@Exceptions raised: (InvenioWebSubmitIconCreatorError)
be raised or propagated by this function when the icon creation process
fails for one reason or another.
"""
## SANITY CHECKS:
## Does the options dictionary contain all expected keys?
##
## A list of the names of the expected options:
expected_option_names
=
[
'input-file'
,
\
'icon-name'
,
\
'icon-file-format'
,
\
'multipage-icon'
,
\
'multipage-icon-delay'
,
\
'icon-scale'
,
\
'verbosity'
]
expected_option_names
.
sort
()
## A list of the option names that have been received:
received_option_names
=
options
.
keys
()
received_option_names
.
sort
()
if
expected_option_names
!=
received_option_names
:
## Error: he dictionary of options had an illegal structure:
msg
=
"""Error: Unexpected value received for "options" parameter."""
raise
InvenioWebSubmitIconCreatorError
(
msg
)
## Do we have an input file to work on?
if
options
[
"input-file"
]
in
(
None
,
""
):
## No input file - stop the icon creation:
msg
=
"Error: unable to determine the name of the file from which "
\
"the icon is to be created."
raise
InvenioWebSubmitIconCreatorError
(
msg
)
else
:
## Get the file type of the input file:
tmp_file_extension
=
options
[
"input-file"
]
.
split
(
"."
)[
-
1
]
if
tmp_file_extension
.
lower
()
not
in
CFG_ALLOWED_FILE_EXTENSIONS
:
## Ilegal input file type.
msg
=
"Error: icons can be only be created from
%s
files, "
\
"not [
%s
]."
%
(
str
(
CFG_ALLOWED_FILE_EXTENSIONS
),
\
tmp_file_extension
.
lower
())
raise
InvenioWebSubmitIconCreatorError
(
msg
)
else
:
subject_filetype
=
tmp_file_extension
.
lower
()
## Wash the requested icon name:
if
type
(
options
[
"icon-name"
])
is
not
str
:
options
[
"icon-name"
]
=
""
else
:
(
dummy
,
name_iconfile
)
=
os
.
path
.
split
(
options
[
"icon-name"
])
if
name_iconfile
!=
""
:
## Take just the basename component of the icon file:
options
[
"icon-name"
]
=
name_iconfile
## Do we have an icon file format?
icon_format
=
options
[
"icon-file-format"
]
if
icon_format
in
(
None
,
""
):
## gif by default:
options
[
"icon-file-format"
]
=
"gif"
elif
str
(
icon_format
)
.
lower
()
not
in
CFG_ALLOWED_FILE_EXTENSIONS
:
## gif if an invalid icon type was supplied:
options
[
"icon-file-format"
]
=
"gif"
else
:
## Use the provided icon type:
options
[
"icon-file-format"
]
=
icon_format
.
lower
()
## Wash the use-first-page-only flag according to the type of the
## requested icon:
if
options
[
"icon-file-format"
]
!=
"gif"
:
## Since the request icon isn't a gif file, it can't be animated
## and should be created from the first "page" of the original file:
options
[
"multipage-icon"
]
=
False
else
:
## The requested icon is a gif. Verify that the multipage-icon
## flag is a boolean value. If not, set it to False by default:
if
type
(
options
[
"multipage-icon"
])
is
not
bool
:
## Non-boolean value: default to False:
options
[
"multipage-icon"
]
=
False
## Wash the delay time for frames in an animated gif icon:
if
type
(
options
[
"multipage-icon-delay"
])
is
not
int
:
## Invalid value - set it to default:
options
[
"multipage-icon-delay"
]
=
100
elif
options
[
"multipage-icon-delay"
]
<
0
:
## Can't have negative delays:
options
[
"multipage-icon-delay"
]
=
100
## Wash the icon scaling information:
if
not
re_imagemagic_scale_parameter_format
.
match
(
options
[
"icon-scale"
]):
## Ivalid value - set it to default:
options
[
"icon-scale"
]
=
"180"
## OK. Begin the icon creation process:
##
## Create a working directory for the icon creation process and get the
## full path to it:
path_workingdir
=
create_working_directory
()
## Copy the file from which the icon is to be created into the
## working directory:
try
:
basename_source_file
=
\
copy_file_to_directory
(
options
[
"input-file"
],
path_workingdir
)
except
IOError
,
err
:
## Unable to copy the source file to the working directory.
msg
=
"Icon creation failed: unable to copy the source image file "
\
"to the working directory. Got this error: [
%s
]"
%
str
(
err
)
raise
InvenioWebSubmitIconCreatorError
(
msg
)
## Create the icon and get its name:
icon_name
=
build_icon
(
path_workingdir
,
\
basename_source_file
,
\
subject_filetype
,
\
options
[
"icon-name"
],
\
options
[
"icon-file-format"
],
\
options
[
"multipage-icon"
],
\
options
[
"multipage-icon-delay"
],
\
options
[
"icon-scale"
])
## Return a tuple containing the working directory and the name of the
## icon file to the caller:
return
(
path_workingdir
,
icon_name
)
## ***** Functions Specific to CLI calling of the program: *****
def
usage
(
wmsg
=
""
,
err_code
=
0
):
"""Print a "usage" message (along with an optional additional warning/error
message) to stderr and exit with a given error code.
@param wmsg: (string) - some kind of warning message for the user.
@param err_code: (integer) - an error code to be passed to sys.exit,
which is called after the usage message has been printed.
@return: None.
"""
## Wash the warning message:
if
wmsg
!=
""
:
wmsg
=
wmsg
.
strip
()
+
"
\n
"
## The usage message:
msg
=
""" Usage:
python ~invenio/lib/python/invenio/websubmit_icon_creator.py \\
[options] input-file.jpg
websubmit_icon_creator.py is used to create an icon for an image.
Options:
-h, --help Print this help.
-V, --version Print version information.
-v, --verbose=LEVEL Verbose level (0=min, 1=default, 9=max).
[NOT IMPLEMENTED]
-s, --icon-scale
Scaling information for the icon that is to
be created. Must be an integer. Defaults to
180.
-m, --multipage-icon
A flag to indicate that the icon should
consist of multiple pages. Will only be
respected if the requested icon type is GIF
and the input file is a PS or PDF consisting
of several pages.
-d, --multipage-icon-delay=VAL
If the icon consists of several pages and is
an animated GIF, a delay between frames can
be specified. Must be an integer. Defaults
to 100.
-f, --icon-file-format=FORMAT
The file format of the icon to be created.
Must be one of:
[pdf, gif, jpg, jpeg, ps, png, bmp]
Defaults to gif.
-o, --icon-name=XYZ
The optional name to be given to the created
icon file. If this is omitted, the icon file
will be given the same name as the input
file, but will be prefixed by "icon-";
Examples:
python ~invenio/lib/python/invenio/websubmit_icon_creator.py \\
--icon-scale=200 \\
--icon-name=test-icon \\
--icon-file-format=jpg \\
test-image.jpg
python ~invenio/lib/python/invenio/websubmit_icon_creator.py \\
--icon-scale=200 \\
--icon-name=test-icon2 \\
--icon-file-format=gif \\
--multipage-icon \\
--multipage-icon-delay=50 \\
test-image2.pdf
"""
sys
.
stderr
.
write
(
wmsg
+
msg
)
sys
.
exit
(
err_code
)
def
get_cli_options
():
"""From the options and arguments supplied by the user via the CLI,
build a dictionary of options to drive websubmit-icon-creator.
For reference, the CLI options available to the user are as follows:
-h, --help -> Display help/usage message and exit;
-V, --version -> Display version information and exit;
-v, --verbose= -> Set verbosity level (0=min, 1=default,
9=max).
-s, --icon-scale -> Scaling information for the icon that
is to be created. Must be of
type 'geometry', as understood
by ImageMagick (Eg. 320 or
320x240 or 100>). Defaults to
180.
-m, --multipage-icon -> A flag to indicate that the icon should
consist of multiple pages. Will only be
respected if the requested icon type is
GIF and the input file is a PS or PDF
consisting of several pages.
-d, --multipage-icon-delay= -> If the icon consists of several pages
and is an animated GIF, a delay between
frames can be specified. Must be an
integer. Defaults to 100.
-f, --icon-file-format= -> The file format of the icon to be
created. Must be one of:
[pdf, gif, jpg, jpeg, ps, png, bmp]
Defaults to gif.
-o, --icon-name= -> The optional name to be given to the
created icon file. If this is omitted,
the icon file will be given the same
name as the input file, but will be
prefixed by "icon-";
@return: (dictionary) of input options and flags, set as
appropriate. The dictionary has the following structure:
+ input-file: (string) - the path to the input file (i.e. that
which is to be stamped;
+ icon-name: (string) - the name of the icon that is to be created
by the program. This is optional - if not provided,
a default name will be applied to the icon file instead;
+ multipage-icon: (boolean) - used only when the original file
is a PDF or PS file. If False, the created icon will feature ONLY
the first page of the PDF. If True, ALL pages of the PDF will
be included in the created icon. Note: If the icon type is not
gif, this flag will be forced as False.
+ multipage-icon-delay: (integer) - used only when the original
file is a PDF or PS AND use-first-page-only is False AND
the icon type is gif.
This allows the user to specify the delay between "pages"
of a multi-page (animated) icon.
+ icon-scale: (integer) - the scaling information to be used for the
creation of the new icon.
+ icon-file-format: (string) - the file format of the icon that is
to be created. Legal values are:
[pdf, gif, jpg, jpeg, ps, png, bmp]
+ verbosity: (integer) - the verbosity level under which the program
is to run;
So, an example of the returned dictionary could be something like:
{ 'input-file' : "demo-picture-file.jpg",
'icon-name' : "icon-demo-picture-file",
'icon-file-format' : "gif",
'multipage-icon' : True,
'multipage-icon-delay' : 100,
'icon-scale' : 180,
'verbosity' : 0,
}
"""
## dictionary of important values relating to cli call of program:
options
=
{
'input-file'
:
""
,
'icon-name'
:
""
,
'icon-file-format'
:
""
,
'multipage-icon'
:
False
,
'multipage-icon-delay'
:
100
,
'icon-scale'
:
180
,
'verbosity'
:
0
,
}
## Get the options and arguments provided by the user via the CLI:
try
:
myoptions
,
myargs
=
getopt
.
getopt
(
sys
.
argv
[
1
:],
"hVv:s:md:f:o:"
,
\
[
"help"
,
"version"
,
"verbosity="
,
"icon-scale="
,
"multipage-icon"
,
"multipage-icon-delay="
,
"icon-file-format="
,
"icon-name="
])
except
getopt
.
GetoptError
,
err
:
## Invalid option provided - usage message
usage
(
wmsg
=
"Error:
%(msg)s
."
%
{
'msg'
:
str
(
err
)
})
## Get the input file from the arguments list (it should be the
## first argument):
if
len
(
myargs
)
>
0
:
options
[
"input-file"
]
=
myargs
[
0
]
## Extract the details of the options:
for
opt
in
myoptions
:
if
opt
[
0
]
in
(
"-V"
,
"--version"
):
## version message and exit
sys
.
stdout
.
write
(
"
%s
\n
"
%
__revision__
)
sys
.
stdout
.
flush
()
sys
.
exit
(
0
)
elif
opt
[
0
]
in
(
"-h"
,
"--help"
):
## help message and exit
usage
()
elif
opt
[
0
]
in
(
"-v"
,
"--verbosity"
):
## Get verbosity level:
if
not
opt
[
1
]
.
isdigit
():
options
[
'verbosity'
]
=
0
elif
int
(
opt
[
1
])
not
in
xrange
(
0
,
10
):
options
[
'verbosity'
]
=
0
else
:
options
[
'verbosity'
]
=
int
(
opt
[
1
])
elif
opt
[
0
]
in
(
"-o"
,
"--icon-name"
):
## Get the name of the icon that is to be created:
options
[
"icon-name"
]
=
opt
[
1
]
elif
opt
[
0
]
in
(
"-f"
,
"--icon-file-format"
):
## The file format of the icon file:
if
str
(
opt
[
1
])
.
lower
()
not
in
CFG_ALLOWED_FILE_EXTENSIONS
:
## Illegal file format requested for icon:
usage
()
else
:
## gif if an invalid icon type was supplied:
options
[
"icon-file-format"
]
=
str
(
opt
[
1
])
.
lower
()
elif
opt
[
0
]
in
(
"-m"
,
"--multipage-icon"
):
## The user would like a multipage (animated) icon:
options
[
'multipage-icon'
]
=
True
elif
opt
[
0
]
in
(
"-d"
,
"--multipage-icon-delay"
):
## The delay to be used in the case of a multipage (animated) icon:
try
:
frame_delay
=
int
(
opt
[
1
])
except
ValueError
:
## Invalid value for delay supplied. Usage message.
usage
()
else
:
if
frame_delay
>=
0
:
options
[
'multipage-icon-delay'
]
=
frame_delay
elif
opt
[
0
]
in
(
"-s"
,
"--icon-scale"
):
## The scaling information for the icon:
if
re_imagemagic_scale_parameter_format
.
match
(
opt
[
1
]):
options
[
'icon-scale'
]
=
opt
[
1
]
else
:
usage
()
##
## Done. Return the dictionary of options:
return
options
def
create_icon_cli
():
"""The function responsible for triggering the icon creation process when
called via the CLI.
This function will effectively get the CLI options, then pass them to
function that is responsible for coordinating the icon creation process
itself.
Once stamping has been completed, an attempt will be made to copy the
icon file to the current working directory. If this can't be done, the
path to the icon will be printed to stdout instead.
"""
## Get CLI options and arguments:
input_options
=
get_cli_options
()
## Create the icon file and obtain the name of the working directory in
## which the icon file is situated and the name of the icon file:
try
:
(
working_dir
,
icon_file
)
=
create_icon
(
input_options
)
except
InvenioWebSubmitIconCreatorError
,
err
:
## Something went wrong:
sys
.
stderr
.
write
(
"Icon creation failed: [
%s
]
\n
"
%
str
(
err
))
sys
.
stderr
.
flush
()
sys
.
exit
(
1
)
if
not
os
.
access
(
"./
%s
"
%
icon_file
,
os
.
F_OK
):
## Copy the icon file into the current directory:
try
:
shutil
.
copyfile
(
"
%s
/
%s
"
%
(
working_dir
,
icon_file
),
\
"./
%s
"
%
icon_file
)
except
IOError
:
## Report that it wasn't possible to copy the icon file locally
## and offer the user a path to it:
msg
=
"It was not possible to copy the icon file to the "
\
"current working directory.
\n
You can find it here: "
\
"[
%s
/
%s
].
\n
"
\
%
(
working_dir
,
icon_file
)
sys
.
stderr
.
write
(
msg
)
sys
.
stderr
.
flush
()
else
:
## A file exists in curdir with the same name as the final icon file.
## Just print out a message stating this fact, along with the path to
## the icon file in the temporary working directory:
msg
=
"The icon file [
%s
] has not been copied to the current "
\
"working directory because a file with this name already "
\
"existed there.
\n
You can find the icon file here: "
\
"[
%s
/
%s
].
\n
"
%
(
icon_file
,
working_dir
,
icon_file
)
sys
.
stderr
.
write
(
msg
)
sys
.
stderr
.
flush
()
## Start proceedings for CLI calls:
if
__name__
==
"__main__"
:
create_icon_cli
()
Event Timeline
Log In to Comment