scifijunkie/pwnagotchi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #713 from alanyee/patch-1

Browse Source 
Replace string formatting with logging laziness in __init__.py
This commit is contained in:
evilsocket 2019-12-10 21:06:57 +02:00 committed by GitHub
commit 93bb633010
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pwnagotchi/__init__.py
View File

@ -1,8 +1,8 @@
import subprocess
import os import os
import logging import logging
import time import time
import re import re
import pwnagotchi.ui.view as view import pwnagotchi.ui.view as view
import pwnagotchi import pwnagotchi
@ -27,17 +27,17 @@ def set_name(new_name):
if new_name != current: if new_name != current:
global _name global _name
logging.info("setting unit hostname '%s' -> '%s'" % (current, new_name)) logging.info("setting unit hostname '%s' -> '%s'", current, new_name)
with open('/etc/hostname', 'wt') as fp: with open('/etc/hostname', 'wt') as fp:
fp.write(new_name) fp.write(new_name)
with open('/etc/hosts', 'rt') as fp: with open('/etc/hosts', 'rt') as fp:
prev = fp.read() prev = fp.read()
logging.debug("old hosts:\n%s\n" % prev) logging.debug("old hosts:\n%s\n", prev)
with open('/etc/hosts', 'wt') as fp: with open('/etc/hosts', 'wt') as fp:
patched = prev.replace(current, new_name, -1) patched = prev.replace(current, new_name, -1)
logging.debug("new hosts:\n%s\n" % patched) logging.debug("new hosts:\n%s\n", patched)
fp.write(patched) fp.write(patched)
os.system("hostname '%s'" % new_name) os.system("hostname '%s'" % new_name)
@ -109,7 +109,7 @@ def shutdown():
def restart(mode): def restart(mode):
logging.warning("restarting in %s mode ..." % mode) logging.warning("restarting in %s mode ...", mode)
if mode == 'AUTO': if mode == 'AUTO':
os.system("touch /root/.pwnagotchi-auto") os.system("touch /root/.pwnagotchi-auto")
@ -123,7 +123,7 @@ def restart(mode):
def reboot(mode=None): def reboot(mode=None):
if mode is not None: if mode is not None:
mode = mode.upper() mode = mode.upper()
logging.warning("rebooting in %s mode ..." % mode) logging.warning("rebooting in %s mode ...", mode)
else: else:
logging.warning("rebooting ...") logging.warning("rebooting ...")