Skip to content

Commit

Permalink
refactoring for cidr_to_netmask (#79)
Browse files Browse the repository at this point in the history
* refactoring for cidr_to_netmask using ipaddress
  • Loading branch information
nichogx authored Jan 29, 2020
1 parent 64646ac commit ee00164
Showing 1 changed file with 3 additions and 38 deletions.
41 changes: 3 additions & 38 deletions protonvpn_cli/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@
import fileinput
import getpass
import random
import ipaddress
# External Libraries
import requests
# ProtonVPN-CLI functions
Expand Down Expand Up @@ -222,44 +223,8 @@ def wait_for_network(wait_time):


def cidr_to_netmask(cidr):
netmask = "0"
cidr_netmask = {
1: "128.0.0.0",
2: "192.0.0.0",
3: "224.0.0.0",
4: "240.0.0.0",
5: "248.0.0.0",
6: "252.0.0.0",
7: "254.0.0.0",
8: "255.0.0.0",
9: "255.128.0.0",
10: "255.192.0.0",
11: "255.224.0.0",
12: "255.240.0.0",
13: "255.248.0.0",
14: "255.252.0.0",
15: "255.254.0.0",
16: "255.255.0.0",
17: "255.255.128.0",
18: "255.255.192.0",
19: "255.255.224.0",
20: "255.255.240.0",
21: "255.255.248.0",
22: "255.255.252.0",
23: "255.255.254.0",
24: "255.255.255.0",
25: "255.255.255.128",
26: "255.255.255.192",
27: "255.255.255.224",
28: "255.255.255.240",
29: "255.255.255.248",
30: "255.255.255.252",
31: "255.255.255.254",
32: "255.255.255.255",
}

netmask = cidr_netmask[cidr]
return netmask
subnet = ipaddress.IPv4Network("0.0.0.0/{0}".format(cidr))
return str(subnet.netmask)


def make_ovpn_template():
Expand Down

0 comments on commit ee00164

Please sign in to comment.