07f58e1cca3aa2aa43e8666b36dfa1182b773036
[redakcja.git] / scripts / fix_pd.py
1 #!/usr/bin/env python
2 # -*- coding: utf-8 -*-
3 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
5 #
6 import sys
7 sys.path.append('.')
8 sys.path.append('./apps')
9 sys.path.append('./lib')
10
11 from django.core.management import setup_environ
12 from redakcja import settings
13
14 setup_environ(settings)
15
16 from catalogue.models import Book, Chunk
17 import re
18
19 fixed = {}
20
21 datepd = r"(<dc:date.pd[^>]+>)([0-9]+)(</dc:date.pd>)"
22 def fix(book, author, dry_run=True):
23     if len(book) == 0:
24         print "%s does not contain chunks" % book.slug
25         return
26     fc = book[0]
27     txt = fc.materialize()
28
29     if dry_run:
30         m = re.search(datepd, txt)
31         if m:
32             print("%s: %s->%d" % (book.slug, m.groups()[1], int(m.groups()[1])+71))
33         else:
34             print("%s: date.pd not found??" % (book.slug,))
35     else:
36         dates = {}
37         def up_date(match):
38             tagopen, date, tagclose = match.groups()
39             olddate=date
40             date = str(int(date)+71)
41             dates['date'] = date
42             dates['olddate'] = olddate
43             dates['overflow'] = False
44             if int(date) > 2012:
45                dates['overflow'] = True
46                return tagopen+date+tagclose
47
48         new_txt = re.sub(datepd, up_date, txt)
49         if dates:
50             print "%s: %s->%s" % (book.slug, dates['olddate'], dates['date'])
51             if dates['overflow']:
52                 print "oops, new date would overfow to the future, i'm not changing"
53                 return  
54            # fc.commit(new_txt, author=author, description=u"Automatyczne poprawienie daty przejścia do domeny publicznej z %s na %s" % (dates['olddate'], dates['date']))
55         else:
56             print "skipping %s" % book.slug
57 import sys
58 import getopt
59 from django.contrib.auth.models import User
60 opts, oth_ = getopt.getopt(sys.argv[1:],[],[ "seriously"])
61 dry_run = not (("--seriously",'') in opts)
62 me = User.objects.get(username='marcinkoziej')
63 if dry_run:
64     print "This is a dry run, to really change dates, run with --seriously"
65 for b in Book.objects.all():
66     fix(b, me, dry_run)
67
68     
69