Files
Bolt/core/utils.py
2019-01-20 14:14:20 +05:30

80 lines
2.1 KiB
Python

import re
from core.config import token
def longestCommonSubstring(s1, s2):
m = [[0] * (1 + len(s2)) for i in range(1 + len(s1))]
longest, x_longest = 0, 0
for x in range(1, 1 + len(s1)):
for y in range(1, 1 + len(s2)):
if s1[x - 1] == s2[y - 1]:
m[x][y] = m[x - 1][y - 1] + 1
if m[x][y] > longest:
longest = m[x][y]
x_longest = x
else:
m[x][y] = 0
return s1[x_longest - longest: x_longest]
def stringToBinary(string):
return ''.join(format(ord(x), 'b') for x in string)
def strength(string):
digits = re.findall(r'\d', string)
lowerAlphas = re.findall(r'[a-z]', string)
upperAlphas = re.findall(r'[A-Z]', string)
entropy = len(set(digits + lowerAlphas + upperAlphas))
if not digits:
entropy = entropy/2
return entropy
def isProtected(parsed):
protected = False
parsedForms = list(parsed.values())
for oneForm in parsedForms:
inputs = oneForm['inputs']
for inp in inputs:
name = inp['name']
kind = inp['type']
value = inp['value']
if re.match(token, value):
protected = True
return protected
def extractHeaders(headers):
sorted_headers = {}
matches = re.findall(r'(.*):\s(.*)', headers)
for match in matches:
header = match[0]
value = match[1]
try:
if value[-1] == ',':
value = value[:-1]
sorted_headers[header] = value
except IndexError:
pass
return sorted_headers
def getUrl(url, data, GET):
if GET:
return url.split('?')[0]
else:
return url
def getParams(url, data, GET):
params = {}
if GET:
if '=' in url:
data = url.split('?')[1]
if data[:1] == '?':
data = data[1:]
else:
data = ''
parts = data.split('&')
for part in parts:
each = part.split('=')
try:
params[each[0]] = each[1]
except IndexError:
params = None
return params