mirror of
				https://github.com/zulip/zulip.git
				synced 2025-11-03 21:43:21 +00:00 
			
		
		
		
	This way, commands with arguments containing whitespace or metacharacters are unambiguously readable. Signed-off-by: Anders Kaseorg <andersk@mit.edu>
		
			
				
	
	
		
			127 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
			
		
		
	
	
			127 lines
		
	
	
		
			3.2 KiB
		
	
	
	
		
			Python
		
	
	
		
			Executable File
		
	
	
	
	
#!/usr/bin/env python3
 | 
						|
 | 
						|
import os
 | 
						|
import platform
 | 
						|
import shlex
 | 
						|
import sys
 | 
						|
import subprocess
 | 
						|
 | 
						|
if False:
 | 
						|
    from typing import Callable, List
 | 
						|
 | 
						|
TOOLS_DIR = os.path.dirname(__file__)
 | 
						|
ROOT_DIR = os.path.dirname(TOOLS_DIR)
 | 
						|
sys.path.insert(0, ROOT_DIR)
 | 
						|
from scripts.lib.zulip_tools import get_dev_uuid_var_path
 | 
						|
UUID_VAR_PATH = get_dev_uuid_var_path()
 | 
						|
 | 
						|
def run(check_func):
 | 
						|
    # type: (Callable[[], bool]) -> None
 | 
						|
    '''
 | 
						|
    This decorator simply runs functions.  It makes it more
 | 
						|
    convenient to add new checks without a big main() function.
 | 
						|
    '''
 | 
						|
    rc = check_func()
 | 
						|
    if not rc:
 | 
						|
        sys.exit(1)
 | 
						|
 | 
						|
def run_command(args):
 | 
						|
    # type: (List[str]) -> None
 | 
						|
    print(' '.join(map(shlex.quote, args)))
 | 
						|
    subprocess.check_call(args)
 | 
						|
 | 
						|
@run
 | 
						|
def check_python_version():
 | 
						|
    # type: () -> bool
 | 
						|
    subprocess.check_call(['/usr/bin/env', 'python', '-V'])
 | 
						|
    return True
 | 
						|
 | 
						|
@run
 | 
						|
def pwd():
 | 
						|
    # type: () -> bool
 | 
						|
    print(os.getcwd())
 | 
						|
    return True
 | 
						|
 | 
						|
@run
 | 
						|
def host_info():
 | 
						|
    # type: () -> bool
 | 
						|
    print(platform.platform())
 | 
						|
    return True
 | 
						|
 | 
						|
@run
 | 
						|
def check_django():
 | 
						|
    # type: () -> bool
 | 
						|
    try:
 | 
						|
        import django
 | 
						|
        print('Django version:', django.get_version())
 | 
						|
        return True
 | 
						|
    except ImportError:
 | 
						|
        print('''
 | 
						|
            ERROR!
 | 
						|
            We cannot import Django, which is usually a
 | 
						|
            symptom of not having your Python venv
 | 
						|
            set up correctly.
 | 
						|
 | 
						|
            Make sure your shell does this at init time:
 | 
						|
 | 
						|
            source /srv/zulip-py3-venv/bin/activate
 | 
						|
 | 
						|
            Or maybe you forget to run inside your VM?
 | 
						|
            ''')
 | 
						|
        return False
 | 
						|
 | 
						|
@run
 | 
						|
def provision_version():
 | 
						|
    # type: () -> bool
 | 
						|
    fn = os.path.join(UUID_VAR_PATH, 'provision_version')
 | 
						|
    with open(fn) as f:
 | 
						|
        version = f.read().strip()
 | 
						|
    print('latest version provisioned:', version)
 | 
						|
    from version import PROVISION_VERSION
 | 
						|
    print('desired version:', PROVISION_VERSION)
 | 
						|
    if version != PROVISION_VERSION:
 | 
						|
        print('You need to provision!')
 | 
						|
        return False
 | 
						|
    return True
 | 
						|
 | 
						|
@run
 | 
						|
def node_stuff():
 | 
						|
    # type: () -> bool
 | 
						|
    print('node version:')
 | 
						|
    subprocess.check_call(['node', '--version'])
 | 
						|
    return True
 | 
						|
 | 
						|
@run
 | 
						|
def test_models():
 | 
						|
    # type: () -> bool
 | 
						|
    settings_module = "zproject.settings"
 | 
						|
    os.environ['DJANGO_SETTINGS_MODULE'] = settings_module
 | 
						|
    import django
 | 
						|
    django.setup()
 | 
						|
    from zerver.models import UserProfile, Realm
 | 
						|
    print('Num realms: ', Realm.objects.count())
 | 
						|
    print('Num users: ', UserProfile.objects.count())
 | 
						|
    return True
 | 
						|
 | 
						|
@run
 | 
						|
def check_venv():
 | 
						|
    # type: () -> bool
 | 
						|
    path = os.path.join(ROOT_DIR, 'scripts', 'lib', 'hash_reqs.py')
 | 
						|
    cache_dir = '/srv/zulip-venv-cache/'
 | 
						|
    for fn in ['dev.txt']:
 | 
						|
        requirements_file = os.path.join(ROOT_DIR, "requirements", fn)
 | 
						|
        output = subprocess.check_output([path, requirements_file], universal_newlines=True)
 | 
						|
        sha1sum = output.split()[0]
 | 
						|
        print(fn, 'venv sha: ', sha1sum)
 | 
						|
        if not os.path.exists(os.path.join(cache_dir, sha1sum)):
 | 
						|
            print('Your venv may be improperly installed!')
 | 
						|
            return False
 | 
						|
    return True
 | 
						|
 | 
						|
@run
 | 
						|
def check_migrations():
 | 
						|
    # type: () -> bool
 | 
						|
    print()
 | 
						|
    rc = subprocess.check_call('./tools/test-migrations')
 | 
						|
    return (rc == 0)
 |