mirror of
https://github.com/sqlmapproject/sqlmap.git
synced 2024-11-25 11:03:47 +03:00
Minor code cleanup
This commit is contained in:
parent
2c4f6d2e99
commit
0a81415f2f
|
@ -22,7 +22,7 @@ __author__ = ('Nicola Larosa <nico-NoSp@m-tekNico.net>,'
|
|||
|
||||
__docformat__ = "restructuredtext en"
|
||||
|
||||
__revision__ = '$Id: odict.py 129 2005-09-12 18:15:28Z teknico $'
|
||||
__revision__ = '$Id$'
|
||||
|
||||
__version__ = '0.2.2'
|
||||
|
||||
|
@ -38,40 +38,40 @@ import types, warnings
|
|||
class OrderedDict(dict):
|
||||
"""
|
||||
A class of dictionary that keeps the insertion order of keys.
|
||||
|
||||
|
||||
All appropriate methods return keys, items, or values in an ordered way.
|
||||
|
||||
|
||||
All normal dictionary methods are available. Update and comparison is
|
||||
restricted to other OrderedDict objects.
|
||||
|
||||
|
||||
Various sequence methods are available, including the ability to explicitly
|
||||
mutate the key ordering.
|
||||
|
||||
|
||||
__contains__ tests:
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3),))
|
||||
>>> 1 in d
|
||||
1
|
||||
>>> 4 in d
|
||||
0
|
||||
|
||||
|
||||
__getitem__ tests:
|
||||
|
||||
|
||||
>>> OrderedDict(((1, 3), (3, 2), (2, 1)))[2]
|
||||
1
|
||||
>>> OrderedDict(((1, 3), (3, 2), (2, 1)))[4]
|
||||
Traceback (most recent call last):
|
||||
KeyError: 4
|
||||
|
||||
|
||||
__len__ tests:
|
||||
|
||||
|
||||
>>> len(OrderedDict())
|
||||
0
|
||||
>>> len(OrderedDict(((1, 3), (3, 2), (2, 1))))
|
||||
3
|
||||
|
||||
|
||||
get tests:
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.get(1)
|
||||
3
|
||||
|
@ -81,9 +81,9 @@ class OrderedDict(dict):
|
|||
5
|
||||
>>> d
|
||||
OrderedDict([(1, 3), (3, 2), (2, 1)])
|
||||
|
||||
|
||||
has_key tests:
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.has_key(1)
|
||||
1
|
||||
|
@ -95,11 +95,11 @@ class OrderedDict(dict):
|
|||
"""
|
||||
Create a new ordered dictionary. Cannot init from a normal dict,
|
||||
nor from kwargs, since items order is undefined in those cases.
|
||||
|
||||
|
||||
If the ``strict`` keyword argument is ``True`` (``False`` is the
|
||||
default) then when doing slice assignment - the ``OrderedDict`` you are
|
||||
assigning from *must not* contain any keys in the remaining dict.
|
||||
|
||||
|
||||
>>> OrderedDict()
|
||||
OrderedDict([])
|
||||
>>> OrderedDict({1: 1})
|
||||
|
@ -282,7 +282,7 @@ class OrderedDict(dict):
|
|||
def __repr__(self):
|
||||
"""
|
||||
Used for __repr__ and __str__
|
||||
|
||||
|
||||
>>> r1 = repr(OrderedDict((('a', 'b'), ('c', 'd'), ('e', 'f'))))
|
||||
>>> r1
|
||||
"OrderedDict([('a', 'b'), ('c', 'd'), ('e', 'f')])"
|
||||
|
@ -320,7 +320,7 @@ class OrderedDict(dict):
|
|||
>>> d[1:3] = OrderedDict(((1, 2), (5, 6), (7, 8)))
|
||||
>>> d
|
||||
OrderedDict([(0, 1), (1, 2), (5, 6), (7, 8), (3, 4)])
|
||||
|
||||
|
||||
>>> a = OrderedDict(((0, 1), (1, 2), (2, 3)), strict=True)
|
||||
>>> a[3] = 4
|
||||
>>> a
|
||||
|
@ -344,12 +344,12 @@ class OrderedDict(dict):
|
|||
>>> a[::-1] = OrderedDict([(0, 1), (1, 2), (2, 3), (3, 4)])
|
||||
>>> a
|
||||
OrderedDict([(3, 4), (2, 3), (1, 2), (0, 1)])
|
||||
|
||||
|
||||
>>> d = OrderedDict([(0, 1), (1, 2), (2, 3), (3, 4)])
|
||||
>>> d[:1] = 3
|
||||
Traceback (most recent call last):
|
||||
TypeError: slice assignment requires an OrderedDict
|
||||
|
||||
|
||||
>>> d = OrderedDict([(0, 1), (1, 2), (2, 3), (3, 4)])
|
||||
>>> d[:1] = OrderedDict([(9, 8)])
|
||||
>>> d
|
||||
|
@ -443,7 +443,7 @@ class OrderedDict(dict):
|
|||
def __getattr__(self, name):
|
||||
"""
|
||||
Implemented so that access to ``sequence`` raises a warning.
|
||||
|
||||
|
||||
>>> d = OrderedDict()
|
||||
>>> d.sequence
|
||||
[]
|
||||
|
@ -462,7 +462,7 @@ class OrderedDict(dict):
|
|||
def __deepcopy__(self, memo):
|
||||
"""
|
||||
To allow deepcopy to work with OrderedDict.
|
||||
|
||||
|
||||
>>> from copy import deepcopy
|
||||
>>> a = OrderedDict([(1, 1), (2, 2), (3, 3)])
|
||||
>>> a['test'] = {}
|
||||
|
@ -491,7 +491,7 @@ class OrderedDict(dict):
|
|||
"""
|
||||
``items`` returns a list of tuples representing all the
|
||||
``(key, value)`` pairs in the dictionary.
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.items()
|
||||
[(1, 3), (3, 2), (2, 1)]
|
||||
|
@ -504,7 +504,7 @@ class OrderedDict(dict):
|
|||
def keys(self):
|
||||
"""
|
||||
Return a list of keys in the ``OrderedDict``.
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.keys()
|
||||
[1, 3, 2]
|
||||
|
@ -514,10 +514,10 @@ class OrderedDict(dict):
|
|||
def values(self, values=None):
|
||||
"""
|
||||
Return a list of all the values in the OrderedDict.
|
||||
|
||||
|
||||
Optionally you can pass in a list of values, which will replace the
|
||||
current list. The value list must be the same len as the OrderedDict.
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.values()
|
||||
[3, 2, 1]
|
||||
|
@ -595,7 +595,7 @@ class OrderedDict(dict):
|
|||
def pop(self, key, *args):
|
||||
"""
|
||||
No dict.pop in Python 2.2, gotta reimplement it
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.pop(3)
|
||||
2
|
||||
|
@ -627,7 +627,7 @@ class OrderedDict(dict):
|
|||
"""
|
||||
Delete and return an item specified by index, not a random one as in
|
||||
dict. The index is -1 by default (the last item).
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.popitem()
|
||||
(2, 1)
|
||||
|
@ -673,7 +673,7 @@ class OrderedDict(dict):
|
|||
def update(self, from_od):
|
||||
"""
|
||||
Update from another OrderedDict or sequence of (key, value) pairs
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 0), (0, 1)))
|
||||
>>> d.update(OrderedDict(((1, 3), (3, 2), (2, 1))))
|
||||
>>> d
|
||||
|
@ -705,11 +705,11 @@ class OrderedDict(dict):
|
|||
def rename(self, old_key, new_key):
|
||||
"""
|
||||
Rename the key for a given value, without modifying sequence order.
|
||||
|
||||
|
||||
For the case where new_key already exists this raise an exception,
|
||||
since if new_key exists, it is ambiguous as to what happens to the
|
||||
associated values, and the position of new_key in the sequence.
|
||||
|
||||
|
||||
>>> od = OrderedDict()
|
||||
>>> od['a'] = 1
|
||||
>>> od['b'] = 2
|
||||
|
@ -741,10 +741,10 @@ class OrderedDict(dict):
|
|||
def setitems(self, items):
|
||||
"""
|
||||
This method allows you to set the items in the dict.
|
||||
|
||||
|
||||
It takes a list of tuples - of the same sort returned by the ``items``
|
||||
method.
|
||||
|
||||
|
||||
>>> d = OrderedDict()
|
||||
>>> d.setitems(((3, 1), (2, 3), (1, 2)))
|
||||
>>> d
|
||||
|
@ -759,10 +759,10 @@ class OrderedDict(dict):
|
|||
``setkeys`` all ows you to pass in a new list of keys which will
|
||||
replace the current set. This must contain the same set of keys, but
|
||||
need not be in the same order.
|
||||
|
||||
|
||||
If you pass in new keys that don't match, a ``KeyError`` will be
|
||||
raised.
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.keys()
|
||||
[1, 3, 2]
|
||||
|
@ -790,9 +790,9 @@ class OrderedDict(dict):
|
|||
"""
|
||||
You can pass in a list of values, which will replace the
|
||||
current list. The value list must be the same len as the OrderedDict.
|
||||
|
||||
|
||||
(Or a ``ValueError`` is raised.)
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.setvalues((1, 2, 3))
|
||||
>>> d
|
||||
|
@ -812,7 +812,7 @@ class OrderedDict(dict):
|
|||
def index(self, key):
|
||||
"""
|
||||
Return the position of the specified key in the OrderedDict.
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.index(3)
|
||||
1
|
||||
|
@ -825,10 +825,10 @@ class OrderedDict(dict):
|
|||
def insert(self, index, key, value):
|
||||
"""
|
||||
Takes ``index``, ``key``, and ``value`` as arguments.
|
||||
|
||||
|
||||
Sets ``key`` to ``value``, so that ``key`` is at position ``index`` in
|
||||
the OrderedDict.
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.insert(0, 4, 0)
|
||||
>>> d
|
||||
|
@ -849,7 +849,7 @@ class OrderedDict(dict):
|
|||
def reverse(self):
|
||||
"""
|
||||
Reverse the order of the OrderedDict.
|
||||
|
||||
|
||||
>>> d = OrderedDict(((1, 3), (3, 2), (2, 1)))
|
||||
>>> d.reverse()
|
||||
>>> d
|
||||
|
@ -860,10 +860,10 @@ class OrderedDict(dict):
|
|||
def sort(self, *args, **kwargs):
|
||||
"""
|
||||
Sort the key order in the OrderedDict.
|
||||
|
||||
|
||||
This method takes the same arguments as the ``list.sort`` method on
|
||||
your version of Python.
|
||||
|
||||
|
||||
>>> d = OrderedDict(((4, 1), (2, 2), (3, 3), (1, 4)))
|
||||
>>> d.sort()
|
||||
>>> d
|
||||
|
@ -875,7 +875,7 @@ class Keys(object):
|
|||
# FIXME: should this object be a subclass of list?
|
||||
"""
|
||||
Custom object for accessing the keys of an OrderedDict.
|
||||
|
||||
|
||||
Can be called like the normal ``OrderedDict.keys`` method, but also
|
||||
supports indexing and sequence methods.
|
||||
"""
|
||||
|
@ -896,7 +896,7 @@ class Keys(object):
|
|||
"""
|
||||
You cannot assign to keys, but you can do slice assignment to re-order
|
||||
them.
|
||||
|
||||
|
||||
You can only do slice assignment if the new set of keys is a reordering
|
||||
of the original set.
|
||||
"""
|
||||
|
@ -966,7 +966,7 @@ class Keys(object):
|
|||
class Items(object):
|
||||
"""
|
||||
Custom object for accessing the items of an OrderedDict.
|
||||
|
||||
|
||||
Can be called like the normal ``OrderedDict.items`` method, but also
|
||||
supports indexing and sequence methods.
|
||||
"""
|
||||
|
@ -1076,7 +1076,7 @@ class Items(object):
|
|||
class Values(object):
|
||||
"""
|
||||
Custom object for accessing the values of an OrderedDict.
|
||||
|
||||
|
||||
Can be called like the normal ``OrderedDict.values`` method, but also
|
||||
supports indexing and sequence methods.
|
||||
"""
|
||||
|
@ -1098,7 +1098,7 @@ class Values(object):
|
|||
def __setitem__(self, index, value):
|
||||
"""
|
||||
Set the value at position i to value.
|
||||
|
||||
|
||||
You can only do slice assignment to values if you supply a sequence of
|
||||
equal length to the slice you are replacing.
|
||||
"""
|
||||
|
@ -1167,12 +1167,12 @@ class SequenceOrderedDict(OrderedDict):
|
|||
"""
|
||||
Experimental version of OrderedDict that has a custom object for ``keys``,
|
||||
``values``, and ``items``.
|
||||
|
||||
|
||||
These are callable sequence objects that work as methods, or can be
|
||||
manipulated directly as sequences.
|
||||
|
||||
|
||||
Test for ``keys``, ``items`` and ``values``.
|
||||
|
||||
|
||||
>>> d = SequenceOrderedDict(((1, 2), (2, 3), (3, 4)))
|
||||
>>> d
|
||||
SequenceOrderedDict([(1, 2), (2, 3), (3, 4)])
|
||||
|
@ -1292,7 +1292,7 @@ class SequenceOrderedDict(OrderedDict):
|
|||
>>> d.values = (1, 2, 3)
|
||||
>>> d
|
||||
SequenceOrderedDict([(1, 1), (2, 2), (3, 3)])
|
||||
|
||||
|
||||
>>> d = SequenceOrderedDict(((1, 2), (2, 3), (3, 4)))
|
||||
>>> d
|
||||
SequenceOrderedDict([(1, 2), (2, 3), (3, 4)])
|
||||
|
|
|
@ -22,7 +22,7 @@ def get_pagerank(url):
|
|||
if rank == '':
|
||||
rank = '0'
|
||||
return rank
|
||||
|
||||
|
||||
def int_str(string, integer, factor):
|
||||
for i in range(len(string)) :
|
||||
integer *= factor
|
||||
|
|
|
@ -297,7 +297,7 @@ def checkSqlInjection(place, parameter, value):
|
|||
boundPayload = agent.prefixQuery(sndPayload, prefix, where, clause)
|
||||
boundPayload = agent.suffixQuery(boundPayload, comment, suffix, where)
|
||||
cmpPayload = agent.payload(place, parameter, newValue=boundPayload, where=where)
|
||||
|
||||
|
||||
return cmpPayload
|
||||
|
||||
# Useful to set kb.matchRatio at first based on
|
||||
|
|
|
@ -678,7 +678,7 @@ class Agent:
|
|||
|
||||
limitedQuery += "NOT IN (%s" % (limitStr % num)
|
||||
limitedQuery += "%s %s%s)" % (field if not uniqueField else uniqueField, fromFrom, (" ORDER BY %s" % uniqueField) if uniqueField else "")
|
||||
|
||||
|
||||
|
||||
if orderBy:
|
||||
limitedQuery += orderBy
|
||||
|
|
|
@ -37,7 +37,7 @@ def tableExists(tableFile, regex=None):
|
|||
|
||||
infoMsg = "checking table existence using items from '%s'" % tableFile
|
||||
logger.info(infoMsg)
|
||||
|
||||
|
||||
infoMsg = "adding words used on web page to the check list"
|
||||
logger.info(infoMsg)
|
||||
pageWords = getPageTextWordsSet(kb.originalPage)
|
||||
|
|
|
@ -133,7 +133,7 @@ class Google:
|
|||
raise sqlmapConnectionException, errMsg
|
||||
|
||||
self.__matches = self.__parsePage(page)
|
||||
|
||||
|
||||
if not self.__matches and "detected unusual traffic" in page:
|
||||
warnMsg = "Google has detected 'unusual' traffic from "
|
||||
warnMsg += "this computer disabling further searches"
|
||||
|
|
Loading…
Reference in New Issue
Block a user