Revision: 5962
Author: jayvdb
Date: 2008-10-14 03:07:44 +0000 (Tue, 14 Oct 2008)
Log Message:
-----------
Verify djvu file exists and detect index page name.
Modified Paths:
--------------
trunk/pywikipedia/djvutext.py
Modified: trunk/pywikipedia/djvutext.py
===================================================================
--- trunk/pywikipedia/djvutext.py 2008-10-14 00:19:59 UTC (rev 5961)
+++ trunk/pywikipedia/djvutext.py 2008-10-14 03:07:44 UTC (rev 5962)
@@ -161,6 +161,7 @@
def main():
+ import os
index = None
djvu = None
pages = None
@@ -183,16 +184,33 @@
else:
print "Unknown argument %s" % arg
+ # Check the djvu file exists
+ os.stat(djvu)
+
+ if not index:
+ import os.path
+ index = os.path.basename(djvu)
+
if djvu and index:
- index_page = wikipedia.Page(wikipedia.getSite(), index)
- if not index_page.exists():
- wikipedia.output("%s does not exist" % index)
- raise Exception
+ site = wikipedia.getSite()
+ index_page = wikipedia.Page(site, index)
- wikipedia.output("uploading text from %s to %s" % (djvu, index) )
+ if site.family.name != 'wikisource':
+ raise wikipedia.PageNotFound("Found family '%s'; Wikisource
required.")
+ if not index_page.exists() and index_page.namespace() == 0:
+ index_namespace = wikipedia.Page(site, 'MediaWiki:Proofreadpage index
namespace').get()
+
+ index_page = wikipedia.Page(wikipedia.getSite(),
+ u"%s:%s" % (index_namespace, index))
+
+ if not index_page.exists():
+ raise wikipedia.NoPage("Page '%s' does not exist" % index)
+
+ wikipedia.output("uploading text from %s to %s" % (djvu, index_page) )
+
bot = DjVuTextBot(djvu, index, pages)
- bot.ask = ask
+ bot.ask = ask
bot.run()
else:
wikipedia.showHelp()