• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django-grpc-master
  • .github
    • ISSUE_TEMPLATE.md
  • .circleci
    • config.yml
  • requirements_test.txt
  • Makefile
  • .coveragerc
  • django_grpc_testtools
    • executor.py
    • __init__.py
    • context.py
  • LICENSE
  • manage.py
  • CONTRIBUTING.md
  • .editorconfig
  • AUTHORS.md
  • setup.py
  • django_grpc
    • apps.py
    • __version__.py
    • models.py
    • signals
      • wrapper.py
      • __init__.py
    • __init__.py
    • utils.py
    • serializers
      • __init__.py
      • base.py
    • management
      • commands
        • grpcserver.py
      • __init__.py
  • runtests.py
  • setup.cfg
  • .travis.yml
  • README.md
  • tests
    • test_utils.py
    • manage.py
    • test_signals.py
    • urls.py
    • sampleapp
      • servicer.py
      • helloworld_pb2_grpc.py
      • __init__.py
      • utils.py
      • helloworld_pb2.py
    • test_server.py
    • __init__.py
    • settings.py
    • helpers.py
    • conftest.py
  • requirements.txt
  • .gitignore
  • docs
    • history.rst
    • Makefile
    • make.bat
    • usage.rst
    • readme.rst
    • authors.rst
    • index.rst
    • conf.py
    • contributing.rst
    • installation.rst
  • MANIFEST.in
  • tox.ini
  • requirements_dev.txt

import datetime
import time
import grpc
import sys

from django.core.management.base import BaseCommand
from concurrent import futures
from django.utils import autoreload

from django_grpc.utils import add_servicers, extract_handlers, create_server


class Command(BaseCommand):
    help = 'Run gRPC server'

    def add_arguments(self, parser):
        parser.add_argument('--max_workers', type=int, help="Number of workers")
        parser.add_argument('--port', type=int, default=50051, help="Port number to listen")
        parser.add_argument('--autoreload', action='store_true', default=False)
        parser.add_argument('--list-handlers', action='store_true', default=False, help="Print all registered endpoints")

    def handle(self, *args, **options):
        if options['autoreload'] is True:
            self.stdout.write("ATTENTION! Autoreload is enabled!")
            if hasattr(autoreload, "run_with_reloader"):
                # Django 2.2. and above
                autoreload.run_with_reloader(self._serve, **options)
            else:
                # Before Django 2.2.
                autoreload.main(self._serve, None, options)
        else:
            self._serve(**options)

    def _serve(self, max_workers, port, *args, **kwargs):
        autoreload.raise_last_exception()
        self.stdout.write("Starting server at %s" % datetime.datetime.now())

        server = create_server(max_workers, port)
        server.start()

        self.stdout.write("Server is listening port %s" % port)

        if kwargs['list_handlers'] is True:
            self.stdout.write("Registered handlers:")
            for handler in extract_handlers(server):
                self.stdout.write("* %s" % handler)

        server.wait_for_termination()