#!/usr/bin/env python
import urllib2
import MultipartPostHandler
import cStringIO
#From http://fabien.seisen.org/python/urllib2_multipart.html
from urllib import urlencode
import cookielib
import re
import os
 
class Wiki:
    def __init__(self, domain, path = '/index.php'):
    #def __init__(self, domain, path = '/index.php5'):
        self.domain = domain
        self.path = self.domain + path
        self.token = ''
        self.cookie_processor = urllib2.HTTPCookieProcessor()
        self.opener = urllib2.build_opener(self.cookie_processor, MultipartPostHandler.MultipartPostHandler())
 
    def login(self, user, password):
        #call the login page with no data to get the token, if there is one
        loginpage = self.opener.open(self.path + "?title=Special:UserLogin")
        matches = re.findall('<input type="hidden" name="wpLoginToken" value="(\w*)" /></form>', loginpage.read())
        if matches: self.token = matches[0]
        data = {'wpName': user,
                'wpPassword': password,
                'wpLoginattempt': 'Log in'}
        if self.token: data['wpLoginToken'] = self.token
        url = self.path  + "?title=Special:UserLogin&action=submitlogin&type=login"
        response = self.opener.open(url, urlencode(data))
        return response.read()
 
    def get_image_list(self):
        # Setting to 100,000 by default to try to get all images.
        # Older versions of MediaWiki:
        #image_request = self.opener.open(self.path + "/Special:Imagelist?limit=100000")
        image_request = self.opener.open(self.path + "?limit=100000&ilsearch=&title=Special:ImageList")
        html = image_request.read()
        matches = re.findall("<td class=\"TablePager_col_img_name\"><a href=\".*\" title=\"(.*)\">.*</a> \(<a href=\"(.*)\">file</a>\)</td>", html)
        images = [(match[0],match[1]) for match in matches]
        return images
 
    def get_page_export(self, pages):
        text = "\n".join(pages)
        data = {
            'curonly': 'on',
            'pages': text,
            #'templates': '',
            #'wpDownload': '',
            'submit': 'Export'
        }
        # Older versions of MediaWiki:
        #url = self.path + "/Special:Export"
        url = self.path + "?title=Special:Export"
        result = self.opener.open((url), urlencode(data))
        return result.read()
    def import_pages(self, xml):
        xml_file = open("import.xml", "w")
        xml_file.write(xml)
        xml_file.close()
        xml_file = open("import.xml", "r")
        data = {
            'action': 'submit',
            'xmlimport': xml_file,
            'source': 'upload',
            'submit': 'Upload File'
        }
        url = self.path + "?title=Special:Import&action=submit"
        result = self.opener.open((url), data)
        xml_file.close()
        os.remove("import.xml")
        return result.read()
    def upload_image(self, filename):
        name = filename.split("/")[-1]
        image_file = open(filename, "rb")
        data = {
            'wpUploadFile': image_file,
            'wpDestFile': name,
            'wpUpload': 'Upload File',
            'wpIgnoreWarning': 'off'
        }
        url = self.path + "?title=Special:Upload&action=submit"
        result = self.opener.open((url), data)
        return result.read()
 
 
    def logout(self):
        self.cookie_processor.cookiejar.clear()