mirror of
https://git.wownero.com/lza_menace/wowstash.git
synced 2024-08-15 00:33:15 +00:00
setup docker!
This commit is contained in:
parent
8324dc3444
commit
e260b8f269
4 changed files with 90 additions and 77 deletions
|
@ -10,3 +10,4 @@ flask-login
|
|||
qrcode
|
||||
Pillow
|
||||
git+https://github.com/lalanza808/MoneroPy
|
||||
docker
|
||||
|
|
|
@ -3,6 +3,7 @@ from wowstash.blueprints.meta import meta_bp
|
|||
from wowstash.library.jsonrpc import daemon
|
||||
from wowstash.library.cache import cache
|
||||
from wowstash.library.db import Database
|
||||
from wowstash.library.docker import Docker
|
||||
|
||||
|
||||
@meta_bp.route('/')
|
||||
|
@ -24,8 +25,9 @@ def privacy():
|
|||
@meta_bp.route('/health')
|
||||
def health():
|
||||
return make_response(jsonify({
|
||||
'cache': info.redis.ping(),
|
||||
'db': Database().connected
|
||||
'redis': cache.redis.ping(),
|
||||
'postgres': Database().connected,
|
||||
'docker': Docker().client.ping()
|
||||
}), 200)
|
||||
|
||||
# @app.errorhandler(404)
|
||||
|
|
|
@ -75,81 +75,6 @@ def create_app():
|
|||
from wowstash.library.jsonrpc import from_atomic
|
||||
return from_atomic(a)
|
||||
|
||||
# commands
|
||||
@app.cli.command('create_wallets')
|
||||
def create_wallets():
|
||||
import subprocess
|
||||
from os import makedirs, path
|
||||
from moneropy import account
|
||||
from wowstash import config
|
||||
from wowstash.factory import db
|
||||
from wowstash.models import User
|
||||
from wowstash.library.jsonrpc import daemon
|
||||
|
||||
if not path.isdir(config.WALLET_DIR):
|
||||
makedirs(config.WALLET_DIR)
|
||||
|
||||
wallets_to_create = User.query.filter_by(wallet_created=False)
|
||||
if wallets_to_create:
|
||||
for u in wallets_to_create:
|
||||
print(f'Creating wallet for user {u}')
|
||||
seed, sk, vk, addr = account.gen_new_wallet()
|
||||
command = f"""wownero-wallet-cli \
|
||||
--generate-new-wallet {config.WALLET_DIR}/{u.id}.wallet \
|
||||
--restore-height {daemon.info()['height']} \
|
||||
--password {u.wallet_password} \
|
||||
--mnemonic-language English \
|
||||
--daemon-address {config.DAEMON_PROTO}://{config.DAEMON_HOST}:{config.DAEMON_PORT} \
|
||||
--daemon-login {config.DAEMON_USER}:{config.DAEMON_PASS} \
|
||||
--command version
|
||||
"""
|
||||
proc = subprocess.Popen(command.split(), stdout=subprocess.PIPE)
|
||||
proc.communicate()
|
||||
if proc.returncode == 0:
|
||||
print(f'Successfully created wallet for {u}!')
|
||||
u.wallet_created = True
|
||||
db.session.commit()
|
||||
else:
|
||||
print(f'Failed to create wallet for {u}.')
|
||||
|
||||
@app.cli.command('refresh_wallets')
|
||||
def refresh_wallets():
|
||||
import subprocess
|
||||
from os import kill
|
||||
from moneropy import account
|
||||
from wowstash import config
|
||||
from wowstash.factory import db
|
||||
from wowstash.models import User
|
||||
from wowstash.library.jsonrpc import daemon
|
||||
|
||||
users = User.query.all()
|
||||
for u in users:
|
||||
print(f'Refreshing wallet for {u}')
|
||||
|
||||
if u.wallet_pid is None:
|
||||
break
|
||||
|
||||
# first check if the pid is still there
|
||||
try:
|
||||
kill(u.wallet_pid, 0)
|
||||
except OSError:
|
||||
print('pid does not exist')
|
||||
u.wallet_connected = False
|
||||
u.wallet_pid = None
|
||||
u.wallet_connect_date = None
|
||||
u.wallet_port = None
|
||||
db.session.commit()
|
||||
|
||||
# then check if the user session is still active
|
||||
if u.is_active is False:
|
||||
print('user session inactive')
|
||||
kill(u.wallet_pid, 9)
|
||||
u.wallet_connected = False
|
||||
u.wallet_pid = None
|
||||
u.wallet_connect_date = None
|
||||
u.wallet_port = None
|
||||
db.session.commit()
|
||||
|
||||
# Routes
|
||||
from wowstash.blueprints.auth import auth_bp
|
||||
from wowstash.blueprints.wallet import wallet_bp
|
||||
|
|
85
wowstash/library/docker.py
Normal file
85
wowstash/library/docker.py
Normal file
|
@ -0,0 +1,85 @@
|
|||
from docker import from_env, APIClient
|
||||
from docker.errors import NotFound
|
||||
from socket import socket
|
||||
from wowstash import config
|
||||
from wowstash.models import User
|
||||
from wowstash.library.jsonrpc import daemon
|
||||
|
||||
class Docker(object):
|
||||
def __init__(self):
|
||||
self.client = from_env()
|
||||
self.wownero_image = getattr(config, 'WOWNERO_IMAGE', 'lalanza808/wownero')
|
||||
self.wallet_dir = getattr(config, 'WALLET_DIR', './data/wallets')
|
||||
self.listen_port = 34569
|
||||
|
||||
def create_wallet(self, user_id):
|
||||
u = User.query.get(user_id)
|
||||
command = f"""wownero-wallet-cli \
|
||||
--generate-new-wallet /wallet/{u.id}.wallet \
|
||||
--restore-height {daemon.info()['height']} \
|
||||
--password {u.wallet_password} \
|
||||
--mnemonic-language English \
|
||||
--daemon-address {config.DAEMON_PROTO}://{config.DAEMON_HOST}:{config.DAEMON_PORT} \
|
||||
--daemon-login {config.DAEMON_USER}:{config.DAEMON_PASS} \
|
||||
--log-file /wallet/{u.id}-create.log
|
||||
--command version
|
||||
"""
|
||||
container = self.client.containers.run(
|
||||
self.wownero_image,
|
||||
command=command,
|
||||
auto_remove=True,
|
||||
name=f'create_wallet_{u.id}',
|
||||
remove=True,
|
||||
detach=True,
|
||||
volumes={
|
||||
f'{self.wallet_dir}/{u.id}': {
|
||||
'bind': '/wallet',
|
||||
'mode': 'rw'
|
||||
}
|
||||
}
|
||||
)
|
||||
return container.short_id
|
||||
|
||||
def start_wallet(self, user_id):
|
||||
u = User.query.get(user_id)
|
||||
command = f"""wownero-wallet-rpc \
|
||||
--non-interactive \
|
||||
--rpc-bind-port {self.listen_port} \
|
||||
--wallet-file /wallet/{u.id}.wallet \
|
||||
--rpc-login {u.id}:{u.wallet_password} \
|
||||
--password {u.wallet_password} \
|
||||
--daemon-address {config.DAEMON_PROTO}://{config.DAEMON_HOST}:{config.DAEMON_PORT} \
|
||||
--daemon-login {config.DAEMON_USER}:{config.DAEMON_PASS} \
|
||||
--log-file /wallet/{u.id}-rpc.log
|
||||
"""
|
||||
container = self.client.containers.run(
|
||||
self.wownero_image,
|
||||
command=command,
|
||||
auto_remove=True,
|
||||
name=f'start_wallet_{u.id}',
|
||||
remove=True,
|
||||
detach=True,
|
||||
ports={
|
||||
f'{self.listen_port}/tcp': None
|
||||
},
|
||||
volumes={
|
||||
f'{self.wallet_dir}/{u.id}': {
|
||||
'bind': '/wallet',
|
||||
'mode': 'rw'
|
||||
}
|
||||
}
|
||||
)
|
||||
return container.short_id
|
||||
|
||||
def get_port(self, container_id):
|
||||
client = APIClient()
|
||||
port_data = client.port(container_id, self.listen_port)
|
||||
host_port = port_data[0]['HostPort']
|
||||
return int(host_port)
|
||||
|
||||
def container_exists(self, container_id):
|
||||
try:
|
||||
self.client.containers.get(container_id)
|
||||
return True
|
||||
except NotFound:
|
||||
return False
|
Loading…
Reference in a new issue