Skip to content

Commit

Permalink
Move main() to __main__.
Browse files Browse the repository at this point in the history
Allows "python -m mockupdb".
  • Loading branch information
ajdavis committed Mar 7, 2015
1 parent 25fe7b7 commit 8aa2054
Show file tree
Hide file tree
Showing 2 changed files with 49 additions and 29 deletions.
29 changes: 0 additions & 29 deletions mockupdb/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -1642,32 +1642,3 @@ def interactive_server(port=27017, verbose=True, all_ok=True, name='MockupDB'):
server.autoresponds('buildinfo', version='MockupDB ' + __version__)
server.autoresponds('replSetGetStatus', ok=0)
return server


def main():
"""Start an interactive `MockupDB`.
Use like ``python -m test.mock_mongodb``.
"""
from optparse import OptionParser
parser = OptionParser('Start mock MongoDB server')
parser.add_option('-p', '--port', dest='port', default=27017,
help='port on which mock mongod listens')
parser.add_option('-q', '--quiet',
action='store_false', dest='verbose', default=True,
help="don't print messages to stdout")

options, cmdline_args = parser.parse_args()
if cmdline_args:
parser.error('Unrecognized argument(s): %s' % ' '.join(cmdline_args))

server = interactive_server(port=options.port, verbose=options.verbose)
try:
server.run()
print('Listening on port %d' % server.port)
time.sleep(1e6)
except KeyboardInterrupt:
server.stop()

if __name__ == '__main__':
main()
49 changes: 49 additions & 0 deletions mockupdb/__main__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
# -*- coding: utf-8 -*-
# Copyright 2015 MongoDB, Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""Demonstrate a mocked MongoDB server."""

import time

from mockupdb import interactive_server


def main():
"""Start an interactive `MockupDB`.
Use like ``python -m mockupdb``.
"""
from optparse import OptionParser
parser = OptionParser('Start mock MongoDB server')
parser.add_option('-p', '--port', dest='port', default=27017,
help='port on which mock mongod listens')
parser.add_option('-q', '--quiet',
action='store_false', dest='verbose', default=True,
help="don't print messages to stdout")

options, cmdline_args = parser.parse_args()
if cmdline_args:
parser.error('Unrecognized argument(s): %s' % ' '.join(cmdline_args))

server = interactive_server(port=options.port, verbose=options.verbose)
try:
server.run()
print('Listening on port %d' % server.port)
time.sleep(1e6)
except KeyboardInterrupt:
server.stop()

if __name__ == '__main__':
main()

0 comments on commit 8aa2054

Please sign in to comment.