FilesExpand file tree

 master

/

cli.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

55 lines (40 loc) · 1.51 KB

 master

/

cli.py

Copy path

Top

File metadata and controls

• Code
• Blame

55 lines (40 loc) · 1.51 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

import re

import click

from proxytest import add_from_text_file

from proxytest import test_csv_file

from proxytest import test_single_proxy

def validate_proxy(ctx, param, value):

'''Validate proxy input. The RegEx crudely matches both IPv4 and URLs.'''

validator = re.compile(r'(https|http|socks4|socks5):\/\/'

r'((?:[0-9]{1,3}\.){3}[0-9]{1,3}(:[0-9]{2,5})?'

r'|([\da-z\.-]+)\.([a-z\.]{2,6})([\/\w \.-]*)*\/?)')

if not validator.match(value):

raise click.BadParameter('Please provide a proxy in the format'

'type://address (e.g., https://42.42.42.42)')

else:

return value

@click.group()

def cli():

pass

@cli.command()

@click.argument('proxy', callback=validate_proxy)

@click.option('--iptest', default='iptest.ingokleiber.de',

help='iptest address')

@click.option('--csv', default='proxies.csv', help='CSV path')

def single(proxy, iptest, csv):

test_single_proxy(proxy, iptest, csv)

@cli.command()

@click.argument('csv')

@click.option('--iptest', default='iptest.ingokleiber.de',

help='iptest address')

def csv_file(iptest, csv):

test_csv_file(iptest, csv)

@cli.command()

@click.argument('txt')

@click.option('--iptest', default='iptest.ingokleiber.de',

help='iptest address')

@click.option('--csv', default='proxies.csv', help='CSV path')

def add_from_txt_file(iptest, txt, csv):

add_from_text_file(iptest, txt, csv)

if __name__ == '__main__':

cli()