fah-control/FAHControl

#!/usr/bin/env python2
'''
  Folding@Home Client Control (FAHControl)
  Copyright (C) 2010-2018 Stanford University

  This program is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program.  If not, see <http://www.gnu.org/licenses/>.
'''

import os
import sys
import inspect
import socket
from optparse import OptionParser
from fah import FAHControl, load_fahcontrol_db
from fah.util import *
from fah.db import *


def set_proc_name(name):
    from ctypes import cdll, byref, create_string_buffer

    libc = cdll.LoadLibrary('libc.so.6')
    buff = create_string_buffer(len(name)+1)
    buff.value = name
    libc.prctl(15, byref(buff), 0, 0, 0)


if sys.platform.startswith('linux'): set_proc_name('FAHControl')

# If present, remove the Launch Services -psn_xxx_xxx argument
if len(sys.argv) > 1 and sys.argv[1][:4] == '-psn':
    del sys.argv[1]

parser = OptionParser(usage = 'Usage: %prog [options]')

parser.add_option('--exit', help = 'Tell the running application to exit',
                  action = 'store_true', dest = 'exit')
options, args = parser.parse_args()

# Tell app to exit
if options.exit:
    try:
        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sock.connect(single_app_addr)
        sock.send('EXIT')
        if sock.recv(1024).strip() == 'OK': print ('Ok')
    except Exception as e:
        pass

    sys.exit(0)

# Add executable path to PATH
if getattr(sys, 'frozen', False): path = os.path.dirname(sys.executable)
else: path = os.path.dirname(__file__)
path = os.path.realpath(path)
os.environ['PATH'] = path + os.pathsep + os.environ['PATH']

# Load Glade
dir = os.path.dirname(inspect.getfile(inspect.currentframe()))
if not dir: dir = '.'
glade = dir + '/fah/FAHControl.glade'

if os.path.exists(glade): app = FAHControl(glade)
else:
    from fah.FAHControl_glade import glade_data
    app = FAHControl(glade_data)

try:
    app.run()
except Exception as e:
    print (e)