[Pywikipedia-l] SVN: [5962] trunk/pywikipedia/djvutext.py

[jayvdb at svn.wikimedia.org]

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()