Homec4science
Diffusion invenio-infoscience (iugw2017) Browse

R3600/modules/websubmit/lib/functions/Report_Number_Generation.py
b5d2e83b4469iugw2017

Last Change
Raw File

Report_Number_Generation.py
View Options

## $Id$
## 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.
"""Description: function Report_Number_Generation
This function creates a reference for the submitted
document and saves it in the specified file.
Author: T.Baron"""
__revision__ = "$Id$"
import os
import re
import time
from invenio.config import counters
from invenio.websubmit_config import InvenioWebSubmitFunctionError
def Report_Number_Generation(parameters,curdir,form):
"""Description: function Report_Number_Generation
This function creates a reference for the submitted
document and saves it in the specified file.
Author: T.Baron
PARAMETERS: edsrn: name of the file in which the reference is saved
autorngen: one of "A", "N", "Y"
"A": The reference is the submission number
"N": The reference is taken from a file [edsrn]
"Y": The reference is generated
rnin: name of the file containing the category
counterpath: path to the counter file note you can use:
<PA>yy</PA> to include year
<PA>categ</PA> to include category of the
submission
<PA>file[re]:name_of_file[regular expression to match]</PA> first line of file
generated by submission, matching [re]
<PA>file*[re]:name_of_file [regular expression to match]</PA> all the lines of
a file genereated during submission, matching [re]
separated by - (dash) char .
rnformat: format for the generated reference. You can use:
<PA>yy</PA> to include year
<PA>categ</PA> to include category of the
submission
<PA>file[re]:name_of_file[regular expression to match]</PA> first line of file
generated by submission, matching [re]
<PA>file*[re]:name_of_file [regular expression to match]</PA> all the lines of
a file genereated during submission, matching [re]
separated by - (dash) char .
yeargen: if "AUTO", current year, else the year is
extracted from the file [yeargen]
"""
global doctype,access,act,dir,rn
# The program must automatically generate the report number
# Generate Year
if parameters['autorngen'] == "Y":
if parameters['yeargen'] == "AUTO":
# Current year is used
yy = time.strftime("%Y")
else :
# If yeargen != auto then the contents of the file named 'yeargen' is used
# Assumes file uses DD-MM-YYYY format
fp = open("%s/%s" % (curdir,parameters['yeargen']),"r")
mydate = fp.read()
fp.close()
yy = re.sub("^......","",mydate)
# Evaluate category - Category is read from the file named 'rnin
if os.path.isfile("%s/%s" % (curdir,parameters['rnin'])):
fp = open("%s/%s" % (curdir,parameters['rnin']),"r")
category = fp.read()
category = category.replace("\n","")
else:
category = ""
def get_pa_tag_content(pa_content):
"""Get content for <PA>XXX</PA>.
@param pa_content: MatchObject for <PA>(.*)</PA>.
return: if pa_content=yy => 4 digits year
if pa_content=categ =>category
if pa_content=file[re]:a_file => first line of file a_file matching re
if pa_content=file*p[re]:a_file => all lines of file a_file, matching re,
separated by - (dash) char.
"""
pa_content=pa_content.groupdict()['content']
sep = '-'
out = ''
if pa_content=='yy':
out = yy
elif pa_content=='categ':
out = category
elif pa_content.startswith('file'):
filename = ""
with_regexp = 0
regexp = ""
if "[" in pa_content:
with_regexp = 1
split_index_start = pa_content.find("[")
split_index_stop = pa_content.rfind("]")
regexp = pa_content[split_index_start+1:split_index_stop]
filename = pa_content[split_index_stop+2:]#]:
else :
filename = pa_content.split(":")[1]
if os.path.exists(os.path.join(curdir, filename)):
fp = open(os.path.join(curdir, filename), 'r')
if pa_content[:5]=="file*":
out = sep.join(map(lambda x: re.split(regexp,x.strip())[-1], fp.readlines()))
else:
out = re.split(regexp, fp.readline().strip())[-1]
fp.close()
return out
counter_path = re.sub('<PA>(?P<content>[^<]*)</PA>',
get_pa_tag_content,
parameters['counterpath'])
counter_path = counter_path.replace(" ","")
counter_path = counter_path.replace("\n","")
rn_format = re.sub('<PA>(?P<content>[^<]*)</PA>',
get_pa_tag_content,
parameters['rnformat'])
# Check if the report number does not already exists
if os.path.exists("%s/%s" % (curdir, parameters['edsrn'])):
fp = open("%s/%s" % (curdir,parameters['edsrn']),"r")
oldrn = fp.read()
fp.close()
if oldrn != "" and not re.search("\?\?\?",oldrn):
rn = oldrn
return ""
# create it
rn = Create_Reference(counter_path, rn_format)
rn = rn.replace("\n","")
rn = rn.replace("\r","")
rn = rn.replace("\015","")
rn = rn.replace("\013","")
rn = rn.replace("\012","")
# The file edsrn is created in the submission directory, and it stores the report number
fp = open("%s/%s" % (curdir,parameters['edsrn']),"w")
fp.write(rn)
fp.close()
# The report number is just read from a specified file
elif parameters['autorngen'] == "N":
fp = open("%s/%s" % (curdir,parameters['edsrn']),"r")
rn = fp.read()
fp.close()
# Some documents are really annoying and insist on a totally different way of doing things
# This code is for documents which have the access number in the report
# number (instead of using a counter)
elif parameters['autorngen'] == "A":
rn = parameters['rnformat'].replace ("<PA>access</PA>",access)
# The file accessno/edsrn is created, and it stores the report number
fp = open("%s/%s" % (curdir,parameters['edsrn']),"w")
fp.write(rn)
fp.close()
return ""
def Create_Reference(counter_path,ref_format):
# test if the counters directory exists, if not attempts to create it
if not os.path.exists(counters):
try:
os.mkdir(counters)
except:
raise InvenioWebSubmitFunctionError("File System: Cannot create counters directory %s" % counters)
if not os.path.exists("%s/%s" % (counters,counter_path)):
try:
fp = open("%s/%s" % (counters,counter_path),"w")
except:
raise InvenioWebSubmitFunctionError("File System: no permission to write in counters directory %s" % counters)
fp.write ("0")
fp.close()
# retrieve current counter value
fp = open("%s/%s" % (counters,counter_path),"r")
id = fp.read()
fp.close()
if id == "":
id=0
# increment the counter
id = int(id)+1
# store new value
fp = open("%s/%s" % (counters,counter_path),"w")
fp.write (str(id))
fp.close()
# create final value
reference = "%s-%03d" % (ref_format,id)
# Return the report number prelude with the id concatenated on at the end
return reference
c4science · Help