import random
import urllib.request as urll
import string

FullRandom = False
FourLetters = False
noimage = 'st.prntscr.com/2018/10/13/2048/img/0_173a7b_211be8ff.png'

fullrandomrange=str(string.ascii_lowercase+string.digits)

print('\nFull Random Mode - 1\n\nEnter first 4 letters - 2\n')
decision = input()

while decision == '1':
    print('\nEnterig FullRandom Mode!')
    FullRandom = True
    break
while decision == '2':
    print('\nFirst 4 letters mode chosen:\n\nEnter your first 4 letters:')
    fourlettermode = input()
    print('\nEntering Four Letters Mode')
    FourLetters = True
    break
while decision not in ['1', '2']:
    print('\nRestart the script and enter eiter 1 or 2')
    break


def randomN():
    return random.randrange(0, 9)

def scrapeletters():
    site = 'https://prnt.sc/'+str(fourlettermode)
    n1 = randomN()
    n2 = randomN()
    numbers = str(n1)+str(n2)
    address = ''.join([site,numbers])
    source = urll.urlopen(urll.Request(address, headers={'User-Agent': 'Mozilla/5.0'}))
    html = source.read()

    if noimage in str(html):
        print(address+'\n'+'Invalid URL')
    else:
        print(address+'\n'+'Valid URL')
        file = open('1.txt','a')
        file.write('\n'+address)


def scrapefullrandom():
#while True:
    site = 'https://prnt.sc/'
    fr1 = random.choice(fullrandomrange)
    fr2 = random.choice(fullrandomrange)
    fr3 = random.choice(fullrandomrange)
    fr4 = random.choice(fullrandomrange)
    fr5 = random.choice(fullrandomrange)
    fr6 = random.choice(fullrandomrange)
    fullrandomletters=fr1+fr2+fr3+fr4+fr5+fr6
    address = ''.join([site,fullrandomletters])
    source = urll.urlopen(urll.Request(address, headers={'User-Agent': 'Mozilla/5.0'}))
    html = source.read()

    if noimage in str(html):
        print(address+'\n'+'Invalid URL')
    else:
        print(address+'\n'+'Valid URL')
        file = open('1.txt','a')
        file.write('\n'+address)

while FullRandom == True:
    scrapefullrandom()
else:
    pass
while FourLetters == True:
    scrapeletters()
else:
    pass

'''while __name__ == '__main__':
    scrape()'''