1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (C) 2017 Étienne Loks <etienne.loks_AT_peacefrogsDOTnet>
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
# This program 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 Affero General Public License for more details.
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
# See the file COPYING for details.
from django.core.management.base import BaseCommand
from optparse import make_option
import sys
from ishtar_common.models import DocumentTemplate
from ishtar_common.utils import BColors
try:
input = raw_input
except NameError:
pass
class Command(BaseCommand):
help = "Update ODT templates from v1 to v2"
option_list = BaseCommand.option_list + (
make_option('--quiet',
action='store_true',
dest='quiet',
default=False,
help='Proceed silently with no interactive input.'),
)
def interactive_start(self):
sys.stdout.write(
BColors.HEADER + BColors.BOLD +
"Update ODT templates from v1 to v2\n")
sys.stdout.write(
BColors.ENDC + BColors.WARNING +
"This script need to be run only once. Running it on already "
"migrated ODT files may be a source of error.\n")
sys.stdout.write(BColors.ENDC)
yes = None
while yes != "yes":
sys.stdout.write(
"Are you sure you want to proceed? (yes/[n])\n")
yes = input()
if not yes or yes == "n":
sys.stdout.write(BColors.FAIL + "Aborting\n")
sys.stdout.write(BColors.ENDC)
sys.exit()
def handle(self, *args, **options):
quiet = options['quiet']
if not quiet:
self.interactive_start()
q = DocumentTemplate.objects
nb = q.count()
len_of_nb = str(len(str(nb)))
if not quiet:
sys.stdout.write(BColors.OKGREEN)
errors = []
for idx, document in enumerate(q.all()):
if not quiet:
sys.stdout.write(
("Processing {:" + len_of_nb + "d}/{}\r").format(
idx + 1, nb))
sys.stdout.flush()
try:
document.convert_from_v1()
except IOError as e:
errors.append("Document ({}): ".format(document.pk) +
str(e))
if errors:
sys.stdout.write(BColors.FAIL + "Error while processing:\n")
for error in errors:
sys.stdout.write("* {}\n".format(error))
sys.stdout.write(BColors.ENDC)
print("\n\n")
|
