为Discuz! X3.2批量添加用户的python脚本

为Discuz! X3.2批量添加用户的python脚本

搭建了一个Discuz! X3.2环境用于工作，需要添加用户时没有找到合适的工具，虽然同事最终用按键精灵解决了问题，不过还是决定写一段代码留下来，也许用得上。
之所以是Python而不是PHP，只是最近在接触Python，发现在这种场合更加方便。

添加用户代码：

# encoding: utf-8
'''
Created on 2015年7月1日

@author: ZhongPing
'''
import urllib
import urllib2
import cookielib
import re

class Adder(object):
    '''
    classdocs
    '''
    home_url = ''
    admin_user = ''
    admin_password = ''
    formhash = ''

    def __init__(self, url, admin_user, admin_password):
        '''
        Constructor
        '''
        self.home_url = url + ?
        self.admin_user = admin_user
        self.admin_password = admin_password
        # 初始化一个CookieJar来处理Cookie
        self.cookieJar=cookielib.CookieJar()
        # 实例化一个全局opener
        self.opener=urllib2.build_opener(urllib2.HTTPCookieProcessor(self.cookieJar))
        self.headers ={
            Host:localhost, 
            Referer: url
        }

    def login(self):
        '''
         管理员登录系统
        '''
        # 登陆用户名和密码
        data={
            admin_username:self.admin_user,
            admin_password:self.admin_password,
            'frames':'yes',
            'admin_questionid':'0',
            'submit':'提交'
        }
        # urllib进行编码
        post_data=urllib.urlencode(data)        
        url = self.home_url
        req=urllib2.Request(url,post_data,self.headers)
        result = self.opener.open(req)
        url = self.home_url+'action=members&operation=add'
        req=urllib2.Request(url)
        result = self.opener.open(req)
        tpage = result.read()
        i = tpage.find('')
        match = pattern.match(tpage)
        formhash = ''
        if match:
            formhash = match.groups()[0]
        self.formhash = formhash
        #print(self.formhash)

    def adduser(self,uname,upwd,uemail,ugrpid = '10',emailnotify = '0',addsubmit = '提交'):
        '''
        添加用户
        '''
        url = 
        url = self.home_url+('action=members&operation=add')
        values = {'formhash':self.formhash,
                  'newusername':uname,
                  'newpassword':upwd,
                  'newemail':uemail,
                  'newgroupid':ugrpid,
                  'emailnotify':emailnotify,
                  'addsubmit':addsubmit
        }
        data = urllib.urlencode(values) 
        req=urllib2.Request(url,data,self.headers)
        response = self.opener.open(req)
        the_page = response.read()
        i = the_page.find('

Discuz! 提示

1. 在Main()函数中需要根据实际情况修改相关参数：

file：包括用户信息的文件。
home_url：管理后台的访问路径。
pwd：管理员的访问密码。

2.用户信息文件按顺序存储用户信息（用户名，密码，邮箱，用户组，是否通知，addsubmit）。可以按如下两种方式组织：

包含全部信息：

test1,123456,test1@test.com,10,0,addsubmit
test2,123456,test2@test.com,10,0,addsubmit
test3,123456,test3@test.com,10,0,addsubmit
test4,123456,test4@test.com,10,0,addsubmit

也可以只需要部分信息：

test1,123456,test1@test.com
test2,123456,test2@test.com
test3,123456,test3@test.com
test4,123456,test4@test.com