import os
from flask import request
from flask_restful import Resource
from dashmachine.version import tcmachine_version


class GetVersion(Resource):
    def get(self):
        return {'Version': tcmachine_version}


class ServerShutdown(Resource):
    def get(self):
        os.system('shutdown now')
        return {'Done'}


class ServerReboot(Resource):
    def get(self):
        os.system('reboot')
        return {'Done'}