| 1 | requests |
| 2 | beautifulsoup4 |
| 3 | pycrypto |
steamlogin.py
· 3.1 KiB · Python
Raw
#!/usr/bin/env python
from __future__ import print_function
import sys
try:
import requests
from bs4 import BeautifulSoup as Soup
from Crypto.PublicKey import RSA
from Crypto.Cipher import PKCS1_v1_5
except ImportError:
print("Please run \"pip install requests beautifulsoup4 pycrypto\"", file=sys.stderr)
sys.exit(1)
import time
import json
import base64
import os
H = {"User-Agent": "Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)"}
session = requests.Session()
session.headers.update(H)
yaml = """{3}:
id: "{0}"
token: "{1}"
date: "{2}" """.strip()
def get_key(username):
return session.post("https://steamcommunity.com/login/getrsakey/",
data=dict(username=username, donotcache=time.time()*1000) ).json()
def encode_passwd(key_data, passwd):
mod = int(key_data["publickey_mod"], 16)
exp = int(key_data["publickey_exp"], 16)
rsa = RSA.construct((mod, exp))
cipher = PKCS1_v1_5.new(rsa)
if hasattr(passwd, "encode"):
passwd = passwd.encode("utf-8")
return base64.b64encode(cipher.encrypt(passwd))
def login(username, password):
key = get_key(username)
encrypted_password = encode_passwd(key, password)
return session.post("https://steamcommunity.com/login/dologin/",
data=dict(
username=username,
password=encrypted_password,
emailauth="",
loginfriendlyname="",
captchagid="-1",
captcha_text="",
emailsteamid="",
rsatimestamp=key["timestamp"],
remember_login=True,
donotcache=time.time()*1000
) ).json()
def transfer(url, login_data):
session.post(url,
data=login_data["transfer_parameters"]
)
def get_gameservers():
"""Yields lists containing four items: steam game id, token, date last used and memo"""
page = Soup(session.get("https://steamcommunity.com/dev/managegameservers").text, "html.parser")
response = []
for row in page.findAll("tr")[1:]:
response.append([ i.text.replace("\\", "\\\\").replace('"', '\\"') for i in row.findAll("td")[:4] ])
return response
if __name__ == "__main__":
username = os.environ.get("STEAM_USERNAME")
password = os.environ.get("STEAM_PASSWORD")
if username and password:
if os.path.isfile("/tmp/steam-{}.json".format(username)):
with open("/tmp/steam-{}.json".format(username)) as f:
session.cookies = requests.utils.cookiejar_from_dict(json.loads(f.read()))
else:
login_data = login(username, password)
for url in login_data["transfer_urls"]:
transfer(url, login_data)
for gs in get_gameservers():
if "(expired)" in gs[1]:
gs[1] = gs[1][:32]
print(yaml.format(*gs) + "\n expired: true")
else:
print(yaml.format(*gs))
with open("/tmp/steam-{}.json".format(username), "w") as f:
f.write(json.dumps(requests.utils.dict_from_cookiejar(session.cookies)))
else:
print("Please specify environment variables STEAM_USERNAME and STEAM_PASSWORD", file=sys.stderr)
| 1 | #!/usr/bin/env python |
| 2 | from __future__ import print_function |
| 3 | import sys |
| 4 | |
| 5 | try: |
| 6 | import requests |
| 7 | from bs4 import BeautifulSoup as Soup |
| 8 | from Crypto.PublicKey import RSA |
| 9 | from Crypto.Cipher import PKCS1_v1_5 |
| 10 | except ImportError: |
| 11 | print("Please run \"pip install requests beautifulsoup4 pycrypto\"", file=sys.stderr) |
| 12 | sys.exit(1) |
| 13 | |
| 14 | import time |
| 15 | import json |
| 16 | import base64 |
| 17 | import os |
| 18 | |
| 19 | H = {"User-Agent": "Mozilla/4.0 (compatible; MSIE 5.5; Windows NT)"} |
| 20 | session = requests.Session() |
| 21 | session.headers.update(H) |
| 22 | yaml = """{3}: |
| 23 | id: "{0}" |
| 24 | token: "{1}" |
| 25 | date: "{2}" """.strip() |
| 26 | |
| 27 | def get_key(username): |
| 28 | return session.post("https://steamcommunity.com/login/getrsakey/", |
| 29 | data=dict(username=username, donotcache=time.time()*1000) ).json() |
| 30 | |
| 31 | def encode_passwd(key_data, passwd): |
| 32 | mod = int(key_data["publickey_mod"], 16) |
| 33 | exp = int(key_data["publickey_exp"], 16) |
| 34 | rsa = RSA.construct((mod, exp)) |
| 35 | cipher = PKCS1_v1_5.new(rsa) |
| 36 | if hasattr(passwd, "encode"): |
| 37 | passwd = passwd.encode("utf-8") |
| 38 | return base64.b64encode(cipher.encrypt(passwd)) |
| 39 | |
| 40 | def login(username, password): |
| 41 | key = get_key(username) |
| 42 | encrypted_password = encode_passwd(key, password) |
| 43 | return session.post("https://steamcommunity.com/login/dologin/", |
| 44 | data=dict( |
| 45 | username=username, |
| 46 | password=encrypted_password, |
| 47 | emailauth="", |
| 48 | loginfriendlyname="", |
| 49 | captchagid="-1", |
| 50 | captcha_text="", |
| 51 | emailsteamid="", |
| 52 | rsatimestamp=key["timestamp"], |
| 53 | remember_login=True, |
| 54 | donotcache=time.time()*1000 |
| 55 | ) ).json() |
| 56 | |
| 57 | def transfer(url, login_data): |
| 58 | session.post(url, |
| 59 | data=login_data["transfer_parameters"] |
| 60 | ) |
| 61 | |
| 62 | def get_gameservers(): |
| 63 | """Yields lists containing four items: steam game id, token, date last used and memo""" |
| 64 | page = Soup(session.get("https://steamcommunity.com/dev/managegameservers").text, "html.parser") |
| 65 | response = [] |
| 66 | for row in page.findAll("tr")[1:]: |
| 67 | response.append([ i.text.replace("\\", "\\\\").replace('"', '\\"') for i in row.findAll("td")[:4] ]) |
| 68 | return response |
| 69 | |
| 70 | if __name__ == "__main__": |
| 71 | username = os.environ.get("STEAM_USERNAME") |
| 72 | password = os.environ.get("STEAM_PASSWORD") |
| 73 | if username and password: |
| 74 | if os.path.isfile("/tmp/steam-{}.json".format(username)): |
| 75 | with open("/tmp/steam-{}.json".format(username)) as f: |
| 76 | session.cookies = requests.utils.cookiejar_from_dict(json.loads(f.read())) |
| 77 | else: |
| 78 | login_data = login(username, password) |
| 79 | for url in login_data["transfer_urls"]: |
| 80 | transfer(url, login_data) |
| 81 | for gs in get_gameservers(): |
| 82 | if "(expired)" in gs[1]: |
| 83 | gs[1] = gs[1][:32] |
| 84 | print(yaml.format(*gs) + "\n expired: true") |
| 85 | else: |
| 86 | print(yaml.format(*gs)) |
| 87 | with open("/tmp/steam-{}.json".format(username), "w") as f: |
| 88 | f.write(json.dumps(requests.utils.dict_from_cookiejar(session.cookies))) |
| 89 | else: |
| 90 | print("Please specify environment variables STEAM_USERNAME and STEAM_PASSWORD", file=sys.stderr) |