Sourcery refactored master branch

2proc_pmutex.py

@@ -4,9 +4,7 @@
 import sys
 import threading
 
-interested = []
-interested.append(False)
-interested.append(False)
+interested = [False, False]
 
 
 def enter_region(process):
@@ -33,7 +31,7 @@ def leave_region(process):
 
 
 def proc(n):
-    t = int(random.randint(0, 1))
+    t = random.randint(0, 1)
     enter_region(n)
     time.sleep(t)
     leave_region(n)

CVE-2017-7679.py

@@ -21,13 +21,11 @@ def check_url(url, buff=None):
 
     opener = urllib2.build_opener(urllib2.HTTPSHandler(context=ctx))
     if buff:
-        opener.addheaders.append(("Content-Type", "text/html;" + buff))
+        opener.addheaders.append(("Content-Type", f"text/html;{buff}"))
     else:
         opener.addheaders.append(("Content-Type", "text/html;"))
 
-    data = opener.open(url).read()
-
-    return data
+    return opener.open(url).read()
 
 
 s1 = check_url(sys.argv[1])

SIMM_Pin_Recovery.py

@@ -83,21 +83,11 @@ def encode_pin_cmd(self, n, s):
         B = 0x30 + int(N[1])
         C = 0x30 + int(N[2])
         D = 0x30 + int(N[3])
-        E = 0xFF
-        F = 0xFF
-        G = 0xFF
-        H = 0xFF
-
-        if l > 4:
-            E = 0x30 + int(N[4])
-        if l > 5:
-            F = 0x30 + int(N[5])
-        if l > 6:
-            G = 0x30 + int(N[6])
-        if l > 7:
-            H = 0x30 + int(N[7])
-
-        COMM = [
+        E = 0x30 + int(N[4]) if l > 4 else 0xFF
+        F = 0x30 + int(N[5]) if l > 5 else 0xFF
+        G = 0x30 + int(N[6]) if l > 6 else 0xFF
+        H = 0x30 + int(N[7]) if l > 7 else 0xFF
+        return [
             0xA0,
             0x20,
             0x00,
@@ -111,9 +101,7 @@ def encode_pin_cmd(self, n, s):
             F,
             G,
             H,
-        ]  # APDU with command pkt
-
-        return COMM
+        ]
 
     def crack_pin(self):
         """The function just forms a APDU with the pin then sends it to the reader and waits for the status."""
@@ -186,12 +174,7 @@ def crack_pin(self):
     connection = r[0].createConnection()
     connection.connect()
 
-    if args.startpin:
-        # n= 00274710
-        n = int(args.startpin)
-    else:
-        n = 0
-
+    n = int(args.startpin) if args.startpin else 0
     if args.lenght:
         if 4 <= int(args.lenght) <= 8:
             l = int(args.lenght)

TOTP_test2.py

@@ -23,6 +23,6 @@
     secret = row[2]
     issuer = row[6]
 
-    data = "otpauth://totp/%s?secret=%s&issuer=%s" % (account, secret, issuer)
+    data = f"otpauth://totp/{account}?secret={secret}&issuer={issuer}"
     img = qrcode.make(data)
     img.show()

X16Rlike.py

@@ -46,7 +46,7 @@ def round(b, n, s=32):
     try:
         d = h.digest(s)
     except:
-        d = h.digest()[0 : s * 2]
+        d = h.digest()[:s * 2]
     return d
 
 

asus_thermal.py

@@ -10,16 +10,14 @@ def beep(f):
 
 
 def gettemp():
-    fp = open("/sys/class/thermal/thermal_zone0/temp", "r")
-    temp = float(fp.read()) / 1000
-    fp.close()
+    with open("/sys/class/thermal/thermal_zone0/temp", "r") as fp:
+        temp = float(fp.read()) / 1000
     return temp
 
 
 def getfan():
-    fp = open("/sys/devices/platform/asus-nb-wmi/hwmon/hwmon2/fan1_input")
-    rpm = int(fp.read())
-    fp.close()
+    with open("/sys/devices/platform/asus-nb-wmi/hwmon/hwmon2/fan1_input") as fp:
+        rpm = int(fp.read())
     return rpm
 
 

bigquery_batch_execute.py

@@ -40,7 +40,7 @@ def query(SQL, table_dest):
 
 tmp = tmp.split(";")
 for sql in tmp:
-    if sql != "" and sql != "\n":
-        sql = sql + ";"
+    if sql not in ["", "\n"]:
+        sql = f"{sql};"
         print(sql)
         query(sql, table_dest)

binet_oeis_crawl.py

@@ -16,25 +16,25 @@ def binet(a,b,n):
 
 """
 
+
 l = 20
 for i in range(0,l):
-  for j in range(i,l):
-    tmp = []
-    if i != j:
-      for k in range(1,l):
-        try:
-          tmp.append(str(round(binet(i,j,k))))
-        except:
-          pass
-      q = ",".join(tmp)
-      try:
-        res=op.resultEois(q)
-        m = "a: %d, b: %d, Name: A%06d, Desc: %s\n" % (i,j,op.getNumber(res[0]),op.getName(res[0]))
-      except:
-        m = ""
-        pass
-      sys.stderr.write(m)
-      sys.stdout.write(m)
-      sys.stdout.flush()
-      sys.stderr.flush()
+    for j in range(i,l):
+        tmp = []
+        if i != j:
+            for k in range(1,l):
+              try:
+                tmp.append(str(round(binet(i,j,k))))
+              except:
+                pass
+            q = ",".join(tmp)
+            try:
+                res=op.resultEois(q)
+                m = "a: %d, b: %d, Name: A%06d, Desc: %s\n" % (i,j,op.getNumber(res[0]),op.getName(res[0]))
+            except:
+                m = ""
+            sys.stderr.write(m)
+            sys.stdout.write(m)
+            sys.stdout.flush()
+            sys.stderr.flush()
 

bitnodes.py

@@ -21,8 +21,7 @@ def make_request(*args):
 
 
 def getjson(url):
-    jdata = json.loads(make_request(url))
-    return jdata
+    return json.loads(make_request(url))
 
 
 data = getjson(url)

bitshuffle.py

@@ -31,17 +31,16 @@ def Transpose16X16(tmp):
 
 
 def bitshuffle(buff, mode):
-    if len(buff) % 8 == 0:
-        buff2 = ""
-        if mode == 8:
-            for i in xrange(0, len(buff) - 1, 8):
-                buff2 += Transpose8X8(buff[i : i + 8])
-        elif mode == 16:
-            for i in xrange(0, len(buff) - 1, 16):
-                buff2 += Transpose16X16(buff[i : i + 16])
-        return buff2
-    else:
+    if len(buff) % 8 != 0:
         raise Exception("Buffer must be a multiple of 8 bytes")
+    buff2 = ""
+    if mode == 8:
+        for i in xrange(0, len(buff) - 1, 8):
+            buff2 += Transpose8X8(buff[i : i + 8])
+    elif mode == 16:
+        for i in xrange(0, len(buff) - 1, 16):
+            buff2 += Transpose16X16(buff[i : i + 16])
+    return buff2
 
 
 if __name__ == "__main__":

bitsigtest.py

@@ -25,32 +25,24 @@ def download_page(url):
         import urllib.request  # urllib library for Extracting web pages
 
         try:
-            headers = {}
-            headers[
-                "User-Agent"
-            ] = "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/41.0.2228.0 Safari/537.36"
+            headers = {
+                "User-Agent": "Mozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/41.0.2228.0 Safari/537.36"
+            }
             req = urllib.request.Request(url, headers=headers)
             resp = urllib.request.urlopen(req)
-            respData = str(resp.read())
-            # print (len(respData) // 1024),"KB"
-
-            return respData
+            return str(resp.read())
         except Exception as e:
             return None
     else:  # If the Current Version of Python is 2.x
         import urllib2
 
         try:
-            headers = {}
-            headers[
-                "User-Agent"
-            ] = "Mozilla/5.0 (X11; Linux i686) AppleWebKit/537.17 (KHTML, like Gecko) Chrome/24.0.1312.27 Safari/537.17"
+            headers = {
+                "User-Agent": "Mozilla/5.0 (X11; Linux i686) AppleWebKit/537.17 (KHTML, like Gecko) Chrome/24.0.1312.27 Safari/537.17"
+            }
             req = urllib2.Request(url, headers=headers)
             response = urllib2.urlopen(req)
-            page = response.read()
-            # print (len(page) // 1024),"KB"
-
-            return page
+            return response.read()
         except:
             return None
 
@@ -86,10 +78,9 @@ def _get_all_items(page):
         item, end_content = _get_next_item(page)
         if item == "no_links":
             break
-        else:
-            items.append(item)  # Append all the links in the list named 'Links'
-            # time.sleep(0.1)        #Timer could be used to slow down the request for image downloads
-            page = page[end_content:]
+        items.append(item)  # Append all the links in the list named 'Links'
+        # time.sleep(0.1)        #Timer could be used to slow down the request for image downloads
+        page = page[end_content:]
         if len(item) > 10000:
             break
     return items

btcaddrinputget.py

@@ -9,12 +9,9 @@ def getinputs(addr_test, offset):
     )
     f = urllib2.urlopen(req)
 
-    data = ""
     addrs = []
 
-    for line in f:
-        data += line
-
+    data = "".join(f)
     json_obj = json.loads(data)
 
     for item in json_obj["txs"]:
@@ -23,8 +20,7 @@ def getinputs(addr_test, offset):
             if addr != addr_test:
                 addrs.append(addr)
 
-    addrs = sorted(set(addrs))
-    return addrs
+    return sorted(set(addrs))
 
 
 for i in range(0, 800):

check_nmap.py

@@ -14,20 +14,18 @@
 
 
 def loadfile(f):
-    fp = open(f, "r")
-    d = fp.read()
-    fp.close()
+    with open(f, "r") as fp:
+        d = fp.read()
     return d
 
 
 def savefile(f, data):
-    fp = open(f, "w")
-    fp.write(data)
-    fp.close()
+    with open(f, "w") as fp:
+        fp.write(data)
 
 
 def json2file(f, s):
-    json_data = str(json.dumps(s))
+    json_data = json.dumps(s)
     savefile(f, json_data)
 
 
@@ -41,7 +39,7 @@ def file2json(f):
 
 def getold_nm(host):
     try:
-        old_nm = file2json(".check_nmap.py." + host + ".cache")
+        old_nm = file2json(f".check_nmap.py.{host}.cache")
     except:
         old_nm = None
     return old_nm
@@ -79,7 +77,7 @@ def proc_results(nm, old_nm, host, ports):
             if a != b:
                 output += "tcp port: %s, old_state: %s,  state: %s\n" % (port, a, b)
                 num_status = 2
-                output0 += "tcp %s: %s," % (port, b)
+                output0 += f"tcp {port}: {b},"
     else:
         num_status = 3
         output = "Could not get scan results, check your firewall!"

classic_shor.py

@@ -45,7 +45,7 @@ def factor(N, g, P):
 N = mpz(gmpy2.next_prime(2 ** bits) * gmpy2.next_prime((2 ** bits) + offset))
 # N= 1522605027922533360535618378132637429718068114961380688657908494580122963258952897654000350692006139
 
-while len(s) == 0:
+while not s:
     g = random.randint(2, 2 ** bits)
     print("N:", N, "g:", g)
     s = factor(N, g, P=max_p)

compression_test1.py

@@ -82,9 +82,6 @@ def test2(data):
 
 # test(str(bytearray((1024**2)*100)))
 
-data = ""
-for line in fileinput.input():
-    data += line
-
+data = "".join(fileinput.input())
 test(data)
 test2(data)

count_bits_order.py

@@ -4,11 +4,7 @@
 
 
 def count_ones(s):
-    tmp = 0
-    for i in range(0, len(s) - 1):
-        if s[i] == "1":
-            tmp += 1
-    return tmp
+    return sum(1 for i in range(0, len(s) - 1) if s[i] == "1")
 
 
 d = {}
@@ -26,8 +22,6 @@ def count_ones(s):
 
 data = []
 for k in dict(sorted(d.items())[::-1]):
-    for i in d[k]:
-        data.append(i)
-
+    data.extend(iter(d[k]))
 for line in data:
     print(data)