diff --git a/algorithms/Graph.ipynb b/algorithms/Graph.ipynb index 809ac2c32..1b59816c4 100644 --- a/algorithms/Graph.ipynb +++ b/algorithms/Graph.ipynb @@ -164,61 +164,62 @@ ], "source": [ "class Graph:\n", - " def __init__(self,adj_list:dict ={}, name=\"Graph\", directed=False):\n", - " \n", + " def __init__(self, adj_list: dict = {}, name=\"Graph\", directed=False):\n", " self.plt = plt\n", - " \n", + "\n", " self.nx = nx\n", - " #validation\n", - " \n", + " # validation\n", + "\n", " tmp = []\n", " for i in adj_list:\n", " tmp.extend(adj_list[i])\n", "\n", - " \n", - " if(not directed and set(tmp)==set(adj_list.keys())):\n", + " if not directed and set(tmp) == set(adj_list.keys()):\n", " self.adj_list = adj_list\n", " self.directed = directed\n", " self.name = name\n", " else:\n", " print(\"Error : It is not a valid graph\")\n", - " \n", + "\n", " def __str__(self):\n", - " string = \" \"+self.name+\"\\n\" \n", + " string = \" \" + self.name + \"\\n\"\n", " for i in self.adj_list.keys():\n", - " string +=str(i)+\" : [\"+\",\".join([str(j) for j in self.adj_list[i]])+\"]\\n\"\n", + " string += (\n", + " str(i) + \" : [\" + \",\".join([str(j) for j in self.adj_list[i]]) + \"]\\n\"\n", + " )\n", " return string\n", - " \n", - " \n", - " def add_node(self,node):\n", + "\n", + " def add_node(self, node):\n", " if node not in self.adj_list.keys():\n", - " self.adj_list[node]=[]\n", + " self.adj_list[node] = []\n", " else:\n", " print(f\"Warning : the node'{node}' already exists\")\n", - " def add_edge(self,node1,node2):\n", + "\n", + " def add_edge(self, node1, node2):\n", " graph = self.adj_list\n", - " if(node1 in graph and node2 in graph):\n", + " if node1 in graph and node2 in graph:\n", " graph[node1].append(node2)\n", - " if(not self.directed):\n", + " if not self.directed:\n", " graph[node2].append(node1)\n", + "\n", " def get_nodeSet(self):\n", " return list(self.adj_list.keys())\n", + "\n", " def get_edgSet(self):\n", " edgSet = []\n", " for i in self.adj_list.keys():\n", " for j in self.adj_list[i]:\n", - " edgSet.append(tuple([i,j]))\n", + " edgSet.append(tuple([i, j]))\n", " return edgSet\n", - " \n", - " \n", - " def dfs(self,node): \n", - " if(node in self.adj_list.keys()):\n", + "\n", + " def dfs(self, node):\n", + " if node in self.adj_list.keys():\n", " traverse = []\n", " visited = set()\n", " stack = [node]\n", - " while(stack):\n", + " while stack:\n", " current = stack.pop()\n", - " if(current not in visited):\n", + " if current not in visited:\n", " visited.add(current)\n", " traverse.append(current)\n", " stack.extend(self.adj_list[current])\n", @@ -226,54 +227,55 @@ " return traverse\n", " else:\n", " print(f\"Error : node'{node}' does not exist in graph\")\n", - " \n", - " def bfs(self,node):\n", - " if(node in self.adj_list.keys()):\n", + "\n", + " def bfs(self, node):\n", + " if node in self.adj_list.keys():\n", " traverse = []\n", " visited = set()\n", " queue = [node]\n", - " while(queue):\n", + " while queue:\n", " current = queue.pop(0)\n", - " if(current not in visited):\n", + " if current not in visited:\n", " visited.add(current)\n", " traverse.append(current)\n", " queue.extend(self.adj_list[current])\n", " return traverse\n", - " \n", + "\n", " else:\n", " print(f\"Error : node'{node}' does not exist in graph\")\n", "\n", - " def getPath(self, n1,n2, path=[]):\n", - " g= self.adj_list\n", - " path +=[n1]\n", + " def getPath(self, n1, n2, path=[]):\n", + " g = self.adj_list\n", + " path += [n1]\n", " if n2 not in g[n1]:\n", " for i in g[n1]:\n", - " if(i not in path):\n", + " if i not in path:\n", " path1 = self.getPath(i, n2, path.copy())\n", - " if(path1):\n", - " return path1 \n", + " if path1:\n", + " return path1\n", " else:\n", - " return path+[n2]\n", - " \n", + " return path + [n2]\n", + "\n", " def opt_path(self, path):\n", " g = self.adj_list\n", - " l= len(path)\n", + " l = len(path)\n", " path1 = path.copy()\n", - " t=2\n", + " t = 2\n", " for i in range(l):\n", - " for j in range(i+2,l):\n", - " if(abs(j-i)>t):\n", + " for j in range(i + 2, l):\n", + " if abs(j - i) > t:\n", " n1 = path[i]\n", " n2 = path[j]\n", - " if(n2 in g[n1]):\n", - " path1 = path[:i+1]+path[j:]\n", - " t = abs(j-i)\n", - " return path1 \n", - " def sortestPath(self,n1,n2):\n", - " path = self.getPath(n1,n2)\n", - " spath =self.opt_path(path)\n", + " if n2 in g[n1]:\n", + " path1 = path[: i + 1] + path[j:]\n", + " t = abs(j - i)\n", + " return path1\n", + "\n", + " def sortestPath(self, n1, n2):\n", + " path = self.getPath(n1, n2)\n", + " spath = self.opt_path(path)\n", " return spath\n", - " \n", + "\n", " def display(self):\n", " G = self.nx.Graph()\n", " G.add_nodes_from(self.get_nodeSet())\n", @@ -281,15 +283,16 @@ " self.nx.draw_networkx(G)\n", " self.plt.title(self.name)\n", " self.plt.show()\n", + "\n", " def makeTree(self, Root=None):\n", " temp_ls = dict()\n", - " if((Root == None) or (Root not in self.adj_list.keys())):\n", + " if (Root == None) or (Root not in self.adj_list.keys()):\n", " root = list(self.adj_list.keys())[0]\n", " else:\n", " root = Root\n", " temp_ls[root] = [root]\n", " stack = [root]\n", - " \n", + "\n", " def visite(root):\n", " for i in self.adj_list[root]:\n", " if root not in temp_ls:\n", @@ -298,30 +301,30 @@ " stack.append(i)\n", " temp_ls[root].append(i)\n", " visite(i)\n", - " \n", + "\n", " visite(root)\n", - " \n", + "\n", " return temp_ls\n", "\n", + "\n", "adj_list = {\n", - " 0:[3,5,9,10,1],\n", - " 1:[6,7,4,0,10],\n", - " 2:[10,5],\n", - " 3:[0],\n", - " 4:[1,5,8],\n", - " 5:[2,0,4],\n", - " 6:[1],\n", - " 7:[1],\n", - " 8:[4],\n", - " 9:[0],\n", - " 10:[2,0]\n", + " 0: [3, 5, 9, 10, 1],\n", + " 1: [6, 7, 4, 0, 10],\n", + " 2: [10, 5],\n", + " 3: [0],\n", + " 4: [1, 5, 8],\n", + " 5: [2, 0, 4],\n", + " 6: [1],\n", + " 7: [1],\n", + " 8: [4],\n", + " 9: [0],\n", + " 10: [2, 0],\n", "}\n", - "g = Graph(adj_list,name=\"Demo\")\n", + "g = Graph(adj_list, name=\"Demo\")\n", "g.display()\n", "for i in adj_list:\n", - " \n", - " g1 = Graph(g.makeTree(Root=i),name=\"Demo tree_\"+str(i))\n", - " g1.display()\n" + " g1 = Graph(g.makeTree(Root=i), name=\"Demo tree_\" + str(i))\n", + " g1.display()" ] }, { @@ -356,9 +359,9 @@ "outputs": [], "source": [ "print(g)\n", - "g.add_edge(8,10)\n", + "g.add_edge(8, 10)\n", "print(g)\n", - "print(g.sortestPath(8,10))\n", + "print(g.sortestPath(8, 10))\n", "g.display()" ] }, @@ -389,40 +392,44 @@ ], "source": [ "class Tree:\n", - " def __init__(self,data, parent = None):\n", + " def __init__(self, data, parent=None):\n", " self.data = data\n", - " self.childs=[]\n", - " def addChild(self,*nodes):\n", - " childs=[Tree(i, parent=self) for i in nodes]\n", + " self.childs = []\n", + "\n", + " def addChild(self, *nodes):\n", + " childs = [Tree(i, parent=self) for i in nodes]\n", " self.childs.extend(childs)\n", - " if (len(childs)==1):\n", + " if len(childs) == 1:\n", " return childs[0]\n", " return childs\n", - " \n", + "\n", " def getChilds(self):\n", - " return [i.data for i in self.childs]\n", + " return [i.data for i in self.childs]\n", + "\n", " def getElements(self):\n", " elements = [self.data]\n", " for i in self.childs:\n", " elements.extend(i.getElements())\n", " return elements\n", + "\n", " def traverse(self, order=\"pre\"):\n", " ls = []\n", - " if(order==\"pre\"):\n", + " if order == \"pre\":\n", " ls.append(self.data)\n", " for i in self.childs:\n", " ls.extend(i.traverse(order=order))\n", " return ls\n", - " elif(order==\"post\"):\n", + " elif order == \"post\":\n", " for i in self.childs:\n", " ls.extend(i.traverse(order=order))\n", " ls.append(self.data)\n", " return ls\n", - " \n", + "\n", + "\n", "tree = Tree(1)\n", "t2 = tree.addChild(2)\n", - "t2t3 = t2.addChild(2,3,4,5,1,7,9)\n", - "tree.addChild(90,23,41)\n", + "t2t3 = t2.addChild(2, 3, 4, 5, 1, 7, 9)\n", + "tree.addChild(90, 23, 41)\n", "tree.getElements()\n", "# tree.traverse()" ] @@ -464,138 +471,137 @@ "outputs": [], "source": [ "class BTree:\n", - " def __init__(self,data, parent=None):\n", + " def __init__(self, data, parent=None):\n", " self.data = data\n", " self.left = None\n", " self.right = None\n", " self.parent = parent\n", - " \n", - " def getElement(self,idx):\n", - " path=[]\n", - " while(idx>0):\n", - " if(idx%2==1):\n", - " path.append((idx%2))\n", + "\n", + " def getElement(self, idx):\n", + " path = []\n", + " while idx > 0:\n", + " if idx % 2 == 1:\n", + " path.append((idx % 2))\n", " idx //= 2\n", " else:\n", - " path.append((idx%2)+2)\n", - " idx = idx//2 -1\n", + " path.append((idx % 2) + 2)\n", + " idx = idx // 2 - 1\n", " temp = self\n", " while path and temp:\n", " dr = path.pop()\n", - " if(dr==1):\n", + " if dr == 1:\n", " temp = temp.left\n", " else:\n", " temp = temp.right\n", - " if(temp):\n", + " if temp:\n", " return temp\n", " else:\n", " print(\"id does not exists\")\n", - " \n", + "\n", " def getDepth(self):\n", " ld = self.left.getDepth() if self.left else -1\n", " rd = self.right.getDepth() if self.right else -1\n", - " return max(ld,rd)+1\n", + " return max(ld, rd) + 1\n", + "\n", " def isBalanced(self):\n", " ld = self.left.getDepth() if self.left else -1\n", - " rd= self.right.getDepth() if self.right else -1\n", - " if(abs(ld-rd) in [0,1]):\n", + " rd = self.right.getDepth() if self.right else -1\n", + " if abs(ld - rd) in [0, 1]:\n", " lb = self.left.isBalanced() if self.left else True\n", " rb = self.right.isBalanced() if self.right else True\n", - " return bool(lb*rb)\n", - " \n", + " return bool(lb * rb)\n", + "\n", " def getParent(self):\n", " return self.parent\n", - " \n", + "\n", " def getChilds(self):\n", " return [self.left, self.right]\n", - " \n", + "\n", " def getId(self):\n", - " if(self.parent != None):\n", - " if(self == self.parent.left):\n", - " return self.parent.getId()*2 + 1\n", + " if self.parent != None:\n", + " if self == self.parent.left:\n", + " return self.parent.getId() * 2 + 1\n", " else:\n", - " return self.parent.getId()*2 + 2\n", + " return self.parent.getId() * 2 + 2\n", " else:\n", " return 0\n", - " \n", - " def __setattr__(self,key,value):\n", + "\n", + " def __setattr__(self, key, value):\n", " self.__dict__[key] = value\n", - " if(value):\n", - " if(key in [\"left\", \"right\"]):\n", + " if value:\n", + " if key in [\"left\", \"right\"]:\n", " value.parent = self\n", - " \n", - " \n", + "\n", " def getSize(self):\n", " lsize = self.left.getSize() if self.left else 0\n", " rsize = self.right.getSize() if self.right else 0\n", - " return lsize+1+rsize\n", - " \n", + " return lsize + 1 + rsize\n", + "\n", " def addData(self, data, idx=-1):\n", - " if(idx==-1):\n", - " q = [] \n", - " q.append(self) \n", - " while (len(q)): \n", - " temp = q[0] \n", - " q.pop(0) \n", - " if (not temp.left):\n", - " temp.left = BTree(data, parent=temp) \n", + " if idx == -1:\n", + " q = []\n", + " q.append(self)\n", + " while len(q):\n", + " temp = q[0]\n", + " q.pop(0)\n", + " if not temp.left:\n", + " temp.left = BTree(data, parent=temp)\n", " break\n", " else:\n", - " q.append(temp.left) \n", - " if (not temp.right):\n", + " q.append(temp.left)\n", + " if not temp.right:\n", " temp.right = BTree(data, parent=temp)\n", " break\n", " else:\n", - " q.append(temp.right) \n", + " q.append(temp.right)\n", " else:\n", - " dr = idx%2\n", - " if(dr==1):\n", - " ele = self.getElement(idx//2)\n", + " dr = idx % 2\n", + " if dr == 1:\n", + " ele = self.getElement(idx // 2)\n", " ele.left = BTree(data, parent=ele)\n", " else:\n", - " ele = self.getElement(idx//2 -1)\n", + " ele = self.getElement(idx // 2 - 1)\n", " ele.right = BTree(data, parent=ele)\n", + "\n", " def addNode(self, node, idx=-1):\n", - " if(idx==-1):\n", - " q = [] \n", - " q.append(self) \n", - " while (len(q)): \n", - " temp = q[0] \n", - " q.pop(0) \n", - " if (not temp.left):\n", - " temp.left = node \n", + " if idx == -1:\n", + " q = []\n", + " q.append(self)\n", + " while len(q):\n", + " temp = q[0]\n", + " q.pop(0)\n", + " if not temp.left:\n", + " temp.left = node\n", " break\n", " else:\n", - " q.append(temp.left) \n", - " if (not temp.right):\n", + " q.append(temp.left)\n", + " if not temp.right:\n", " temp.right = node\n", " break\n", " else:\n", - " q.append(temp.right) \n", - " \n", - " \n", - " \n", + " q.append(temp.right)\n", + "\n", " def traverse(self, order=\"pre\"):\n", " ls = []\n", - " if(order==\"pre\"):\n", + " if order == \"pre\":\n", " ls.append(self.data)\n", - " if(self.left):\n", + " if self.left:\n", " ls.extend(self.left.traverse(order))\n", - " if(self.right):\n", + " if self.right:\n", " ls.extend(self.right.traverse(order))\n", " return ls\n", - " elif(order==\"post\"):\n", - " if(self.left):\n", + " elif order == \"post\":\n", + " if self.left:\n", " ls.extend(self.left.traverse(order))\n", - " if(self.right):\n", + " if self.right:\n", " ls.extend(self.right.traverse(order))\n", " ls.append(self.data)\n", " return ls\n", - " elif(order==\"in\"):\n", - " if(self.left):\n", + " elif order == \"in\":\n", + " if self.left:\n", " ls.extend(self.left.traverse(order))\n", " ls.append(self.data)\n", - " if(self.right):\n", + " if self.right:\n", " ls.extend(self.right.traverse(order))\n", " return ls" ] @@ -619,14 +625,13 @@ ], "source": [ "bt = BTree(0)\n", - "bt.left=BTree(1)\n", + "bt.left = BTree(1)\n", "bt.right = BTree(2)\n", "bt.left.left = BTree(3)\n", "bt.left.right = BTree(4)\n", "bt.right.left = BTree(5)\n", "\n", "\n", - "\n", "# bt.getElement(5)\n", "bt.traverse(\"post\")" ] @@ -650,7 +655,7 @@ ], "source": [ "btr = BTree(3)\n", - "ls=[5,6,8,11,4,14,9,17]\n", + "ls = [5, 6, 8, 11, 4, 14, 9, 17]\n", "for i in ls:\n", " btr.addData(i)\n", "btr.traverse(order=\"in\")\n", @@ -721,7 +726,7 @@ "btr.getElement(1).getId()\n", "\n", "\n", - "bt.getSize(),btr.getSize()" + "bt.getSize(), btr.getSize()" ] }, { diff --git a/cryptography/extended_eucledian.ipynb b/cryptography/extended_eucledian.ipynb index fe98c2962..67c2b7ffa 100644 --- a/cryptography/extended_eucledian.ipynb +++ b/cryptography/extended_eucledian.ipynb @@ -62,7 +62,8 @@ " return b\n", " return eucld_gcd(b, r)\n", "\n", - "eucld_gcd(252,105)" + "\n", + "eucld_gcd(252, 105)" ] }, { @@ -85,6 +86,7 @@ "source": [ "import numpy as np\n", "\n", + "\n", "def ext_eucld(a, b):\n", " \"\"\"\n", " Computes the extended Euclidean algorithm to find the greatest common divisor (GCD)\n", @@ -124,13 +126,14 @@ " ls = eucld(a, b)\n", " for i in ls:\n", " row = np.append(row, [row[-2] - i * row[-1]], axis=0)\n", - " \n", + "\n", " if swap:\n", " return list(row[-1])[::-1]\n", - " \n", + "\n", " return list(row[-1])\n", "\n", - "ext_eucld(97, 2)\n" + "\n", + "ext_eucld(97, 2)" ] }, { @@ -150,8 +153,8 @@ "source": [ "a = 5\n", "b = 7\n", - "gcd = eucld_gcd(a,b)\n", - "m,n = ext_eucld(a,b)\n", + "gcd = eucld_gcd(a, b)\n", + "m, n = ext_eucld(a, b)\n", "print(f\"a={a}, b={b}, gcd={gcd}, m={m}, n={n}, ma+nb={m*a+n*b}\")" ] }, diff --git a/cryptography/galois_field.ipynb b/cryptography/galois_field.ipynb index ac68caa85..87f45e8b3 100644 --- a/cryptography/galois_field.ipynb +++ b/cryptography/galois_field.ipynb @@ -49,124 +49,128 @@ "def id2bit(ls: list):\n", " \"\"\"\n", " Converts a list of indices into a binary representation (bit vector).\n", - " \n", - " Given a list of indices (ls), this function returns a list of bits where \n", - " the bit positions corresponding to the indices in the list are set to 1, \n", + "\n", + " Given a list of indices (ls), this function returns a list of bits where\n", + " the bit positions corresponding to the indices in the list are set to 1,\n", " and all other positions are set to 0. The resulting list is reversed.\n", - " \n", + "\n", " Args:\n", " ls (list): A list of indices to be converted to bits.\n", - " \n", + "\n", " Returns:\n", " list: A list of bits representing the binary values.\n", " \"\"\"\n", - " if(len(ls) == 0):\n", + " if len(ls) == 0:\n", " return [0, 0, 0, 0, 0, 0, 0, 0] # Return a default 8-bit array\n", " aa = [0 for _ in range(max(ls) + 1)]\n", " for i in ls:\n", " aa[i] = 1\n", " return aa[::-1]\n", "\n", + "\n", "def bit2id(ls: list, log=False):\n", " \"\"\"\n", " Converts a binary list (bit vector) back to a list of indices.\n", - " \n", + "\n", " Given a list of bits (ls), this function returns the indices of the bits\n", " that are set to 1. The binary list is reversed during the conversion.\n", - " \n", + "\n", " Args:\n", " ls (list): A list of bits representing a binary value.\n", " log (bool, optional): Whether to log intermediate steps (default is False).\n", - " \n", + "\n", " Returns:\n", " list: A list of indices where the bits are set to 1.\n", " \"\"\"\n", " ls = ls[::-1]\n", " aa = []\n", - " \n", + "\n", " for i in range(len(ls)):\n", - " if(ls[i] == 1):\n", + " if ls[i] == 1:\n", " aa.append(i)\n", " return aa[::-1]\n", "\n", + "\n", "def bit2mul(a, b, log=False):\n", " \"\"\"\n", " Multiplies two binary numbers represented as lists of bits.\n", - " \n", - " This function multiplies two binary numbers by performing a bitwise \n", + "\n", + " This function multiplies two binary numbers by performing a bitwise\n", " multiplication and addition over Galois Field (GF(2)).\n", - " \n", + "\n", " Args:\n", " a (list): A list of bits representing the first binary number.\n", " b (list): A list of bits representing the second binary number.\n", " log (bool, optional): Whether to log intermediate steps (default is False).\n", - " \n", + "\n", " Returns:\n", " list: The resulting binary number (list of bits).\n", " \"\"\"\n", " ai = bit2id(a)\n", " bi = bit2id(b)\n", " a, b = a[::-1], b[::-1]\n", - " \n", - " if(ai == []):\n", + "\n", + " if ai == []:\n", " return a\n", - " elif(bi == []):\n", + " elif bi == []:\n", " return b\n", - " \n", + "\n", " addn = [[ai[i] + bi[j] for j in range(len(bi))][::-1] for i in range(len(ai))][::-1]\n", " addn = [id2bit(i) for i in addn]\n", - " \n", + "\n", " maxsiz = max([len(i) for i in addn])\n", " for i in range(len(addn)):\n", - " if(len(addn[i]) < maxsiz):\n", + " if len(addn[i]) < maxsiz:\n", " addn[i] = [0 for _ in range(maxsiz - len(addn[i]))] + addn[i]\n", - " \n", + "\n", " smm = []\n", " for i in range(maxsiz):\n", " t = 0\n", " for j in addn:\n", " t += j[i]\n", " smm.append(t % 2)\n", - " \n", + "\n", " return smm\n", "\n", + "\n", "def bit2add(a, b):\n", " \"\"\"\n", " Adds two binary numbers represented as lists of bits (bitwise addition).\n", - " \n", + "\n", " This function adds two binary numbers by performing a bitwise addition over GF(2).\n", - " \n", + "\n", " Args:\n", " a (list): A list of bits representing the first binary number.\n", " b (list): A list of bits representing the second binary number.\n", - " \n", + "\n", " Returns:\n", " list: The resulting binary number after addition (list of bits).\n", " \"\"\"\n", " a, b = list(a), list(b)\n", " a, b = a[::-1], b[::-1]\n", " maxsiz = max(len(a), len(b))\n", - " \n", - " if(len(a) < maxsiz):\n", + "\n", + " if len(a) < maxsiz:\n", " a = a + [0 for _ in range(maxsiz - len(a))]\n", - " if(len(b) < maxsiz):\n", + " if len(b) < maxsiz:\n", " b = b + [0 for _ in range(maxsiz - len(b))]\n", - " \n", + "\n", " smm = []\n", " for i in range(maxsiz):\n", " smm.append((a[i] + b[i]) % 2)\n", - " \n", + "\n", " return smm[::-1]\n", "\n", + "\n", "def bit2str(bit: list):\n", " \"\"\"\n", " Converts a list of bits into a string.\n", - " \n", + "\n", " This function converts a list of binary bits (0s and 1s) into a string of characters.\n", - " \n", + "\n", " Args:\n", " bit (list): A list of bits (0s and 1s).\n", - " \n", + "\n", " Returns:\n", " str: The string representation of the binary bits.\n", " \"\"\"\n", @@ -175,100 +179,105 @@ " s += str(i)\n", " return s\n", "\n", + "\n", "def str2bit(s: str):\n", " \"\"\"\n", " Converts a string of '0's and '1's into a list of bits.\n", - " \n", + "\n", " This function converts a string containing '0's and '1's into a list of integer bits.\n", - " \n", + "\n", " Args:\n", " s (str): A string containing '0's and '1's.\n", - " \n", + "\n", " Returns:\n", " list: A list of bits (integers).\n", - " \n", + "\n", " Raises:\n", " ValueError: If the string contains characters other than '0' and '1'.\n", " \"\"\"\n", - " if(set(s).issubset(set('01'))):\n", + " if set(s).issubset(set(\"01\")):\n", " bit = [int(i) for i in s]\n", " return bit\n", " else:\n", " print(\"bit string should contain 1s and 0s\")\n", "\n", - "def modgf(dsr: list, dnt = [1, 0, 0, 0, 1, 1, 0, 1, 1]):\n", + "\n", + "def modgf(dsr: list, dnt=[1, 0, 0, 0, 1, 1, 0, 1, 1]):\n", " \"\"\"\n", " Performs polynomial division over Galois Field (GF(2)).\n", "\n", " This function divides the binary polynomial `dsr` by the binary polynomial `dnt`\n", " and returns the quotient and remainder.\n", - " \n", + "\n", " Args:\n", " dsr (list): The dividend as a list of bits (binary polynomial).\n", " dnt (list, optional): The divisor as a list of bits (default is a predefined irreducible polynomial).\n", - " \n", + "\n", " Returns:\n", " tuple: The remainder and quotient as lists of bits.\n", " \"\"\"\n", " dsr = bit2id(dsr)\n", " dnt = bit2id(dnt)\n", " qtnt = []\n", - " \n", - " while (len(dnt) != 0 and len(dsr) != 0 and (max(dnt) - max(dsr) >= 0)):\n", + "\n", + " while len(dnt) != 0 and len(dsr) != 0 and (max(dnt) - max(dsr) >= 0):\n", " ml = max(dnt) - max(dsr)\n", " qtnt.append(ml)\n", " plus = id2bit(dnt)\n", " minus = id2bit([ml + i for i in dsr])\n", " rem = bit2add(plus, minus)\n", " dnt = bit2id(rem)\n", - " \n", + "\n", " return id2bit(dnt), id2bit(qtnt)\n", "\n", + "\n", "def ext_eucld(a, b, log=False):\n", " \"\"\"\n", " Extended Euclidean algorithm for binary polynomials.\n", "\n", " This function computes the extended Euclidean algorithm for binary polynomials `a` and `b`,\n", " returning the coefficients of the linear combination of `a` and `b` that equals the greatest common divisor (GCD).\n", - " \n", + "\n", " Args:\n", " a (list): A list of bits representing the first binary polynomial.\n", " b (list): A list of bits representing the second binary polynomial.\n", " log (bool, optional): Whether to log intermediate steps (default is False).\n", - " \n", + "\n", " Returns:\n", " list: The coefficients of the linear combination of `a` and `b` (as lists of bits).\n", " \"\"\"\n", " ai, bi = bit2id(a), bit2id(b)\n", - " if((len(ai) != 0 and len(bi) != 0)):\n", - " if(max(max(ai), max(bi)) == max(bi)):\n", + " if len(ai) != 0 and len(bi) != 0:\n", + " if max(max(ai), max(bi)) == max(bi):\n", " a, b = b, a\n", - " elif(len(ai) == 0 and len(bi) != 0):\n", + " elif len(ai) == 0 and len(bi) != 0:\n", " a, b = b, a\n", - " \n", + "\n", " def eucld(a, b, log=False):\n", " a, b = a[::-1], b[::-1]\n", - " \n", - " if(set(b) == set([0]) or (b[0] == 1 and (set(b[1:]) == set([0])))):\n", + "\n", + " if set(b) == set([0]) or (b[0] == 1 and (set(b[1:]) == set([0]))):\n", " return []\n", - " \n", + "\n", " ls = []\n", - " \n", + "\n", " while not (b[0] == 1 and (set(b[1:]) == set([0]))):\n", " r, idx = modgf(b[::-1], dnt=a[::-1])\n", " r, idx = r[::-1], idx[::-1]\n", - " \n", - " if(set(r) == set([0])):\n", + "\n", + " if set(r) == set([0]):\n", " return ls\n", - " \n", + "\n", " ls.append(idx[::-1])\n", " a = b\n", " b = r\n", " return ls\n", - " \n", - " row = [[[0, 0, 0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0, 0, 0]],\n", - " [[0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 1]]]\n", - " \n", + "\n", + " row = [\n", + " [[0, 0, 0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0, 0, 0]],\n", + " [[0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 1]],\n", + " ]\n", + "\n", " ls = eucld(a, b)\n", " for i in ls:\n", " r10, r11 = row[-1][0], row[-1][1]\n", @@ -277,32 +286,34 @@ " r1 = bit2add(r21, bit2mul(r11, i))\n", " rowl = [r0, r1]\n", " row.append(rowl)\n", - " \n", + "\n", " return row[-1]\n", "\n", + "\n", "def Gfinv(bit, irrpoly=[1, 0, 0, 0, 1, 1, 0, 1, 1]):\n", " \"\"\"\n", " Computes the multiplicative inverse of a binary polynomial over GF(2).\n", - " \n", + "\n", " This function uses the extended Euclidean algorithm to compute the inverse of a binary polynomial `bit`\n", " with respect to a predefined irreducible polynomial `irrpoly`.\n", - " \n", + "\n", " Args:\n", " bit (list): A list of bits representing the binary polynomial to be inverted.\n", " irrpoly (list, optional): The irreducible polynomial used for the field (default is a predefined polynomial).\n", - " \n", + "\n", " Returns:\n", " list: The multiplicative inverse of the polynomial `bit` (list of bits).\n", " \"\"\"\n", - " if(set(bit) == set('0')):\n", - " return '--'\n", - " \n", + " if set(bit) == set(\"0\"):\n", + " return \"--\"\n", + "\n", " ans = ext_eucld(irrpoly, bit)\n", - " ans = ans[-1][-len(bit):]\n", + " ans = ans[-1][-len(bit) :]\n", " return ans\n", "\n", + "\n", "# Example call\n", - "Gfinv([0, 0, 0, 0, 0, 1, 0, 0], irrpoly=[0, 0, 0, 1, 0, 0, 1, 1])\n" + "Gfinv([0, 0, 0, 0, 0, 1, 0, 0], irrpoly=[0, 0, 0, 1, 0, 0, 1, 1])" ] }, { @@ -320,23 +331,23 @@ "metadata": {}, "outputs": [], "source": [ - "def genmapping(n:int, irrpoly):\n", + "def genmapping(n: int, irrpoly):\n", " \"\"\"\n", - " Generates the elements of GF(2^n) and their corresponding multiplicative inverses \n", + " Generates the elements of GF(2^n) and their corresponding multiplicative inverses\n", " based on the provided irreducible polynomial.\n", "\n", " Parameters:\n", - " n (int): The size of the Galois Field (GF(2^n)). Determines the number of elements \n", + " n (int): The size of the Galois Field (GF(2^n)). Determines the number of elements\n", " in the field, which is 2^n.\n", - " irrpoly (list): A list of bits representing the irreducible polynomial used \n", + " irrpoly (list): A list of bits representing the irreducible polynomial used\n", " for the finite field operations (e.g., [1, 0, 0, 1] for x^3 + 1).\n", "\n", " Returns:\n", " tuple: A tuple containing:\n", - " - gf (list): A list of binary strings of length `n`, representing all elements \n", + " - gf (list): A list of binary strings of length `n`, representing all elements\n", " of GF(2^n). The binary strings are padded with leading zeros.\n", - " - invmap (dict): A dictionary mapping the index of each element in `gf` to the \n", - " index of its multiplicative inverse, using the irreducible \n", + " - invmap (dict): A dictionary mapping the index of each element in `gf` to the\n", + " index of its multiplicative inverse, using the irreducible\n", " polynomial for the field.\n", "\n", " Example:\n", @@ -345,24 +356,26 @@ " # invmap will contain a mapping of the inverses for each non-zero element.\n", " \"\"\"\n", " gf = [str(bin(i))[2:] for i in range(2**n)]\n", - " \n", + "\n", " # Ensure each element has length n (pad with leading zeros if necessary)\n", " for i in range(len(gf)):\n", " if len(gf[i]) < n:\n", - " gf[i] = '0' * (n - len(gf[i])) + gf[i]\n", - " \n", + " gf[i] = \"0\" * (n - len(gf[i])) + gf[i]\n", + "\n", " # Create mappings: index -> element (key2ele) and element -> index (ele2key)\n", " key2ele = dict(enumerate(gf))\n", " ele2key = dict([i[::-1] for i in list(enumerate(gf))])\n", - " \n", + "\n", " # Generate the inverse map for all non-zero elements\n", " invmap = dict()\n", " for i in gf:\n", - " if set(i) != set('0'): # Skip zero element\n", + " if set(i) != set(\"0\"): # Skip zero element\n", " inv = bit2str(Gfinv(str2bit(i), irrpoly=irrpoly)) # Find the inverse of i\n", - " invmap[ele2key[i]] = ele2key[inv] # Map the inverse using element-to-key mapping\n", - " \n", - " return gf, invmap\n" + " invmap[ele2key[i]] = ele2key[\n", + " inv\n", + " ] # Map the inverse using element-to-key mapping\n", + "\n", + " return gf, invmap" ] }, { @@ -372,7 +385,7 @@ "metadata": {}, "outputs": [], "source": [ - "gf5,invmap = genmapping(n=5,irrpoly=id2bit([5,2,0]))" + "gf5, invmap = genmapping(n=5, irrpoly=id2bit([5, 2, 0]))" ] }, { @@ -393,7 +406,7 @@ } ], "source": [ - "set(invmap.values())==set(invmap.keys())" + "set(invmap.values()) == set(invmap.keys())" ] }, { @@ -464,15 +477,14 @@ "source": [ "gf28 = [str(bin(i))[2:] for i in range(256)]\n", "for i in range(len(gf28)):\n", - " if(len(gf28[i])<8):\n", - " gf28[i] = '0'*(8-len(gf28[i])) + gf28[i]\n", + " if len(gf28[i]) < 8:\n", + " gf28[i] = \"0\" * (8 - len(gf28[i])) + gf28[i]\n", "\n", "key2ele = dict(enumerate(gf28))\n", "ele2key = dict([i[::-1] for i in list(enumerate(gf28))])\n", "invmap = dict()\n", "for i in gf28:\n", - "\n", - " if(set(i)!=set('0')):\n", + " if set(i) != set(\"0\"):\n", " inv = bit2str(Gfinv(str2bit(i)))\n", " invmap[ele2key[i]] = ele2key[inv]" ] @@ -495,7 +507,7 @@ } ], "source": [ - "set(invmap.values())==set(invmap.keys())" + "set(invmap.values()) == set(invmap.keys())" ] }, { diff --git a/cryptography/lfsr_bit_stream.ipynb b/cryptography/lfsr_bit_stream.ipynb index 948155139..40aa6f448 100644 --- a/cryptography/lfsr_bit_stream.ipynb +++ b/cryptography/lfsr_bit_stream.ipynb @@ -45,16 +45,16 @@ "def id2bit(ls: list):\n", " \"\"\"\n", " Converts a list of indices to a binary representation (bit array).\n", - " \n", - " Given a list of indices, this function creates a binary list where each index in \n", - " the input list is set to 1 in the output list, and all other positions are set to 0. \n", + "\n", + " Given a list of indices, this function creates a binary list where each index in\n", + " the input list is set to 1 in the output list, and all other positions are set to 0.\n", " The output list is then reversed before returning.\n", - " \n", + "\n", " Args:\n", " ls (list): A list of indices where each index will be set to 1 in the output list.\n", - " \n", + "\n", " Returns:\n", - " list: A list of binary values (0s and 1s), where each index in the input list corresponds \n", + " list: A list of binary values (0s and 1s), where each index in the input list corresponds\n", " to a 1 in the output binary list, and all other indices are 0.\n", " \"\"\"\n", " if len(ls) == 0:\n", @@ -68,13 +68,13 @@ "def bit2id(ls: list):\n", " \"\"\"\n", " Converts a binary list (bit array) to a list of indices where the value is 1.\n", - " \n", + "\n", " This function iterates over the binary list and returns a list of indices where the binary value is 1.\n", " The list is reversed before returning.\n", - " \n", + "\n", " Args:\n", " ls (list): A list of binary values (0s and 1s).\n", - " \n", + "\n", " Returns:\n", " list: A list of indices where the corresponding binary value in the input list is 1.\n", " \"\"\"\n", @@ -89,13 +89,13 @@ "def XOR(*args):\n", " \"\"\"\n", " Performs bitwise XOR on a sequence of values.\n", - " \n", - " This function takes any number of arguments and performs the XOR operation iteratively \n", + "\n", + " This function takes any number of arguments and performs the XOR operation iteratively\n", " across all the input values.\n", - " \n", + "\n", " Args:\n", " *args: A sequence of values (typically integers) on which the XOR operation will be applied.\n", - " \n", + "\n", " Returns:\n", " int: The result of applying the XOR operation across all input values.\n", " \"\"\"\n", @@ -108,26 +108,26 @@ "class LFSR:\n", " \"\"\"\n", " A class representing a Linear Feedback Shift Register (LFSR).\n", - " \n", - " This class models an LFSR, which generates a sequence of bits based on an initial state \n", + "\n", + " This class models an LFSR, which generates a sequence of bits based on an initial state\n", " and a feedback polynomial. The LFSR can be clocked to generate subsequent bits in the sequence.\n", - " \n", + "\n", " Attributes:\n", " seq (list): The current state (bit sequence) of the LFSR.\n", " taps (list): The positions of the taps used for feedback calculation.\n", - " \n", + "\n", " Methods:\n", " clock(): Shifts the bits in the LFSR and computes the new bit based on the feedback.\n", " \"\"\"\n", - " \n", + "\n", " def __init__(self, start, poly):\n", " \"\"\"\n", " Initializes an LFSR with a start state and a feedback polynomial.\n", - " \n", + "\n", " Args:\n", " start (list): The initial state of the LFSR, represented as a list of bits (0s and 1s).\n", " poly (list): A list representing the feedback polynomial, with 1s indicating the taps.\n", - " \n", + "\n", " Raises:\n", " ValueError: If the length of the start state does not match the polynomial length minus one.\n", " \"\"\"\n", @@ -140,8 +140,8 @@ " def clock(self):\n", " \"\"\"\n", " Advances the LFSR by one clock cycle.\n", - " \n", - " This method computes the feedback bit by XORing the bits at the tap positions, \n", + "\n", + " This method computes the feedback bit by XORing the bits at the tap positions,\n", " shifts the state, and adds the feedback bit to the beginning of the sequence.\n", " \"\"\"\n", " feedback = XOR(*[self.seq[bit] for bit in self.taps])\n", @@ -151,24 +151,24 @@ "class A51:\n", " \"\"\"\n", " A class representing the A5/1 stream cipher.\n", - " \n", - " A51 is a stream cipher used in GSM encryption. It combines three LFSRs and uses a majority rule \n", + "\n", + " A51 is a stream cipher used in GSM encryption. It combines three LFSRs and uses a majority rule\n", " to control which LFSRs are clocked. The output is the XOR of the last bits of the LFSRs.\n", - " \n", + "\n", " Attributes:\n", " lfsrs (list): A list of LFSR instances.\n", " clock_bits (list): The bit positions used for clocking each LFSR.\n", " lfsr_count (int): The number of LFSRs used in the cipher.\n", - " \n", + "\n", " Methods:\n", " majority(*bits): Computes the majority bit from a list of bits.\n", " clock(): Advances the cipher and returns the next bit of the keystream.\n", " \"\"\"\n", - " \n", + "\n", " def __init__(self, lfsrs, clock_bits):\n", " \"\"\"\n", " Initializes the A51 cipher with a list of LFSRs and their clocking bits.\n", - " \n", + "\n", " Args:\n", " lfsrs (list): A list of LFSR instances used to generate the keystream.\n", " clock_bits (list): A list indicating the bit positions in each LFSR to use for majority voting.\n", @@ -180,13 +180,13 @@ " def majority(self, *bits):\n", " \"\"\"\n", " Computes the majority bit from a sequence of bits.\n", - " \n", - " This method determines the majority (1 or 0) from the given bits. If the number of 1s \n", + "\n", + " This method determines the majority (1 or 0) from the given bits. If the number of 1s\n", " is greater than or equal to half of the number of LFSRs, the majority bit is 1; otherwise, it is 0.\n", - " \n", + "\n", " Args:\n", " *bits: A sequence of bits (typically 0s and 1s) for which the majority is to be determined.\n", - " \n", + "\n", " Returns:\n", " int: The majority bit (0 or 1).\n", " \"\"\"\n", @@ -200,15 +200,17 @@ " def clock(self):\n", " \"\"\"\n", " Advances the A51 cipher by one clock cycle and generates the next keystream bit.\n", - " \n", - " This method computes the majority bit from the specified clocking positions of the LFSRs, \n", - " clocks the LFSRs if necessary, and outputs the XOR of the last bits of each LFSR as the next \n", + "\n", + " This method computes the majority bit from the specified clocking positions of the LFSRs,\n", + " clocks the LFSRs if necessary, and outputs the XOR of the last bits of each LFSR as the next\n", " bit of the keystream.\n", - " \n", + "\n", " Returns:\n", " int: The next bit in the keystream generated by the A51 cipher.\n", " \"\"\"\n", - " majority = self.majority(*[self.lfsrs[i].seq[self.clock_bits[i]] for i in range(self.lfsr_count)])\n", + " majority = self.majority(\n", + " *[self.lfsrs[i].seq[self.clock_bits[i]] for i in range(self.lfsr_count)]\n", + " )\n", " for i in range(self.lfsr_count):\n", " if self.lfsrs[i].seq[self.clock_bits[i]] == majority:\n", " self.lfsrs[i].clock()\n", @@ -224,7 +226,7 @@ "\n", "# Generate a keystream of 10 bits\n", "stream = [a51.clock() for i in range(10)]\n", - "stream\n" + "stream" ] }, { @@ -236,17 +238,18 @@ "source": [ "import os\n", "\n", + "\n", "def write2txt_file(bitstream, filename):\n", " \"\"\"\n", " Writes a bitstream (string of '0's and '1's) to a text file.\n", "\n", " This function opens a text file in append mode and writes the provided bitstream to it.\n", - " \n", + "\n", " Args:\n", " bitstream (str): A string of '0's and '1's representing the bitstream to be written.\n", " filename (str): The path to the text file where the bitstream will be written.\n", " \"\"\"\n", - " with open(filename, 'a') as f: # Open in append mode to continue writing\n", + " with open(filename, \"a\") as f: # Open in append mode to continue writing\n", " f.write(bitstream)\n", "\n", "\n", @@ -254,25 +257,25 @@ " \"\"\"\n", " Writes a bitstream (string of '0's and '1's) to a binary file.\n", "\n", - " This function converts the bitstream into bytes, pads it to ensure it's a multiple of 8 bits, \n", + " This function converts the bitstream into bytes, pads it to ensure it's a multiple of 8 bits,\n", " and then writes it to a binary file in append mode.\n", - " \n", + "\n", " Args:\n", " bitstream (str): A string of '0's and '1's representing the bitstream to be written.\n", " filename (str): The path to the binary file where the bitstream will be written.\n", " \"\"\"\n", " byte_list = []\n", - " \n", + "\n", " # Pad the bitstream if it's not a multiple of 8\n", " padding = (8 - (len(bitstream) % 8)) % 8\n", - " bitstream += '0' * padding # Add extra '0's to make the length a multiple of 8\n", - " \n", + " bitstream += \"0\" * padding # Add extra '0's to make the length a multiple of 8\n", + "\n", " for i in range(0, len(bitstream), 8):\n", - " byte = bitstream[i:i+8]\n", + " byte = bitstream[i : i + 8]\n", " byte_list.append(int(byte, 2)) # Convert 8 bits to an integer (byte)\n", - " \n", + "\n", " # Append the bytes to the binary file\n", - " with open(filename, 'ab') as f: # 'ab' mode to append to the binary file\n", + " with open(filename, \"ab\") as f: # 'ab' mode to append to the binary file\n", " f.write(bytearray(byte_list))\n", "\n", "\n", @@ -280,28 +283,30 @@ " \"\"\"\n", " Generates a keystream using the A51 cipher and writes it to a file.\n", "\n", - " This function initializes the LFSRs based on the provided data, generates a keystream \n", - " using the A51 cipher, and writes the generated bits to a text file or binary file \n", - " in chunks. It keeps track of the current size of the output file and prints progress \n", + " This function initializes the LFSRs based on the provided data, generates a keystream\n", + " using the A51 cipher, and writes the generated bits to a text file or binary file\n", + " in chunks. It keeps track of the current size of the output file and prints progress\n", " at each 10% interval.\n", "\n", " Args:\n", - " data (dict): A dictionary containing information about the LFSRs, including their \n", + " data (dict): A dictionary containing information about the LFSRs, including their\n", " start values, polynomials, and clock positions.\n", - " target_size (int): The target size of the file in bytes. The function will stop once \n", + " target_size (int): The target size of the file in bytes. The function will stop once\n", " this size is reached.\n", " file_path (str): The path to the output file where the generated bitstream will be written.\n", " \"\"\"\n", " # Initialize the LFSRs and A51 cipher\n", " lfsrs = [LFSR(start=i[\"start\"], poly=i[\"poly\"]) for i in data]\n", " a51 = A51(lfsrs=lfsrs, clock_bits=[i[\"clock\"] for i in data])\n", - " \n", + "\n", " current_size = 0\n", " bitstream_chunk = \"\" # Chunk of bits to write periodically\n", - " chunk_size = 10000 # Number of bits to generate at a time (can adjust for performance)\n", + " chunk_size = (\n", + " 10000 # Number of bits to generate at a time (can adjust for performance)\n", + " )\n", " progress_interval = target_size // 10 # 1/10th of the target size (100 MB)\n", " next_progress_checkpoint = progress_interval\n", - " \n", + "\n", " # Generate bits until the target file size is reached\n", " while current_size < target_size:\n", " # Generate bits in chunks\n", @@ -310,17 +315,21 @@ "\n", " # Write the chunk to file\n", " write2txt_file(bitstream_chunk, file_path)\n", - " \n", + "\n", " # Clear the chunk and update the current file size\n", " bitstream_chunk = \"\"\n", " current_size = os.path.getsize(file_path)\n", - " \n", + "\n", " # Check if the file size has crossed the 1/10th checkpoint\n", " if current_size >= next_progress_checkpoint:\n", - " print(f\"File size crossed {round(next_progress_checkpoint / (1024 * 1024), 2)} MB\")\n", - " next_progress_checkpoint += progress_interval # Update to next 10% checkpoint\n", + " print(\n", + " f\"File size crossed {round(next_progress_checkpoint / (1024 * 1024), 2)} MB\"\n", + " )\n", + " next_progress_checkpoint += (\n", + " progress_interval # Update to next 10% checkpoint\n", + " )\n", "\n", - " print(f\"File generation complete: {file_path} (target)\")\n" + " print(f\"File generation complete: {file_path} (target)\")" ] }, { @@ -342,9 +351,9 @@ "data = [\n", " {\"start\": [0, 1, 0, 1, 1], \"poly\": id2bit([5, 2, 0]), \"clock\": 2},\n", " {\"start\": [1, 0, 0, 1, 0], \"poly\": id2bit([5, 4, 3, 1, 0]), \"clock\": 3},\n", - " {\"start\": [0, 1, 1, 0, 0], \"poly\": id2bit([5, 4, 2, 1, 0]), \"clock\": 2}\n", + " {\"start\": [0, 1, 1, 0, 0], \"poly\": id2bit([5, 4, 2, 1, 0]), \"clock\": 2},\n", "]\n", - "gen_bit_stream(data, target_size=1*1024**2, file_path=\"mine_gen_100MB.txt\")" + "gen_bit_stream(data, target_size=1 * 1024**2, file_path=\"mine_gen_100MB.txt\")" ] }, { diff --git a/cryptography/playfire.ipynb b/cryptography/playfire.ipynb index c7f9ad0b8..bde0bc4c1 100644 --- a/cryptography/playfire.ipynb +++ b/cryptography/playfire.ipynb @@ -35,7 +35,7 @@ " \"\"\"\n", " PlayFire class implements the Playfair cipher for encryption and decryption of messages.\n", "\n", - " The Playfair cipher is a digraph substitution cipher that encrypts pairs of letters. It requires a key, which \n", + " The Playfair cipher is a digraph substitution cipher that encrypts pairs of letters. It requires a key, which\n", " is used to create a 6x6 matrix of letters and digits, and processes the message in pairs.\n", "\n", " Attributes:\n", @@ -51,7 +51,7 @@ " decrypt(msg): Decrypts the given encrypted message using the Playfair cipher.\n", " \"\"\"\n", "\n", - " def __init__(self, key, extra='x'):\n", + " def __init__(self, key, extra=\"x\"):\n", " \"\"\"\n", " Initializes the PlayFire cipher with a key and an optional extra character for padding.\n", "\n", @@ -75,9 +75,9 @@ " \"\"\"\n", " keyy = []\n", " for i in key:\n", - " if(i not in keyy):\n", + " if i not in keyy:\n", " keyy.append(i)\n", - " if(len(set(key)) == len(key)):\n", + " if len(set(key)) == len(key):\n", " return key\n", " else:\n", " print(\"key Error\")\n", @@ -95,7 +95,7 @@ " mtrx = []\n", " idx = 0\n", " for i in range(6):\n", - " t1 = xx[idx:idx + 6]\n", + " t1 = xx[idx : idx + 6]\n", " mtrx.append(t1)\n", " idx = idx + 6\n", " return mtrx\n", @@ -113,13 +113,13 @@ " idxs = [6, 6]\n", " for i in range(6):\n", " for j in range(6):\n", - " if(i == 5):\n", + " if i == 5:\n", " i = -1\n", - " if(j == 5):\n", + " if j == 5:\n", " j = -1\n", - " if(pair[0] == self.key_matrix[i][j]):\n", + " if pair[0] == self.key_matrix[i][j]:\n", " idxs[0] = [i, j]\n", - " if(pair[1] == self.key_matrix[i][j]):\n", + " if pair[1] == self.key_matrix[i][j]:\n", " idxs[1] = [i, j]\n", " return idxs\n", "\n", @@ -134,20 +134,29 @@ " str: The encrypted message.\n", " \"\"\"\n", " msg = list(msg.lower())\n", - " if(len(msg) % 2 == 1):\n", + " if len(msg) % 2 == 1:\n", " msg.append(self.extra)\n", " pairs = []\n", " for i in range(0, len(msg), 2):\n", - " pairs.append(msg[i:i + 2])\n", + " pairs.append(msg[i : i + 2])\n", " en_msg = \"\"\n", " for i in pairs:\n", " idxs = self.find_idx(i)\n", - " if(idxs[0][0] == idxs[1][0]):\n", - " en_m = self.key_matrix[idxs[0][0]][idxs[0][1] + 1] + self.key_matrix[idxs[0][0]][idxs[1][1] + 1]\n", - " elif(idxs[0][1] == idxs[1][1]):\n", - " en_m = self.key_matrix[idxs[0][0] + 1][idxs[0][1]] + self.key_matrix[idxs[1][0] + 1][idxs[1][1]]\n", + " if idxs[0][0] == idxs[1][0]:\n", + " en_m = (\n", + " self.key_matrix[idxs[0][0]][idxs[0][1] + 1]\n", + " + self.key_matrix[idxs[0][0]][idxs[1][1] + 1]\n", + " )\n", + " elif idxs[0][1] == idxs[1][1]:\n", + " en_m = (\n", + " self.key_matrix[idxs[0][0] + 1][idxs[0][1]]\n", + " + self.key_matrix[idxs[1][0] + 1][idxs[1][1]]\n", + " )\n", " else:\n", - " en_m = self.key_matrix[idxs[0][0]][idxs[1][1]] + self.key_matrix[idxs[1][0]][idxs[0][1]]\n", + " en_m = (\n", + " self.key_matrix[idxs[0][0]][idxs[1][1]]\n", + " + self.key_matrix[idxs[1][0]][idxs[0][1]]\n", + " )\n", " en_msg += en_m\n", " return en_msg\n", "\n", @@ -162,22 +171,31 @@ " str: The decrypted plaintext message.\n", " \"\"\"\n", " msg = list(msg.lower())\n", - " if(len(msg) % 2 == 1):\n", + " if len(msg) % 2 == 1:\n", " msg.append(self.extra)\n", " pairs = []\n", " for i in range(0, len(msg), 2):\n", - " pairs.append(msg[i:i + 2])\n", + " pairs.append(msg[i : i + 2])\n", " en_msg = \"\"\n", " for i in pairs:\n", " idxs = self.find_idx(i)\n", - " if(idxs[0][0] == idxs[1][0]):\n", - " en_m = self.key_matrix[idxs[0][0]][idxs[0][1] - 1] + self.key_matrix[idxs[0][0]][idxs[1][1] - 1]\n", - " elif(idxs[0][1] == idxs[1][1]):\n", - " en_m = self.key_matrix[idxs[0][0] - 1][idxs[0][1]] + self.key_matrix[idxs[1][0] - 1][idxs[1][1]]\n", + " if idxs[0][0] == idxs[1][0]:\n", + " en_m = (\n", + " self.key_matrix[idxs[0][0]][idxs[0][1] - 1]\n", + " + self.key_matrix[idxs[0][0]][idxs[1][1] - 1]\n", + " )\n", + " elif idxs[0][1] == idxs[1][1]:\n", + " en_m = (\n", + " self.key_matrix[idxs[0][0] - 1][idxs[0][1]]\n", + " + self.key_matrix[idxs[1][0] - 1][idxs[1][1]]\n", + " )\n", " else:\n", - " en_m = self.key_matrix[idxs[0][0]][idxs[1][1]] + self.key_matrix[idxs[1][0]][idxs[0][1]]\n", + " en_m = (\n", + " self.key_matrix[idxs[0][0]][idxs[1][1]]\n", + " + self.key_matrix[idxs[1][0]][idxs[0][1]]\n", + " )\n", " en_msg += en_m\n", - " return en_msg\n" + " return en_msg" ] }, { @@ -225,7 +243,7 @@ } ], "source": [ - "msg = 'hello1234'\n", + "msg = \"hello1234\"\n", "enc = pf.encrypt(msg)\n", "enc" ] @@ -290,7 +308,7 @@ } ], "source": [ - "pf.decrypt(pf.encrypt('r'))" + "pf.decrypt(pf.encrypt(\"r\"))" ] }, {