
from ubersmith.calls import BaseCall, GroupCall
from ubersmith.utils import prepend_base

__all__ = [
    'GetCall',
    'ListCall',
]

_ = prepend_base(__name__.split('.')[-1])


class GetCall(BaseCall):
    method = _('get')
    required_fields = ['device_id']


class ListCall(GroupCall):
    method = _('list')
    rename_fields = {'clientid': 'client_id'}
    int_fields = ['client_id']


class ModuleGraphCall(FileCall):
    method = _('module_graph')

from ubersmith.calls import BaseCall, GroupCall, FileCall
from ubersmith.utils import prepend_base

__all__ = [
    'GetCall',
    'ListCall',
    'ModuleGraphCall',
]

_ = prepend_base(__name__.split('.')[-1])


class GetCall(BaseCall):
    method = _('get')
    required_fields = ['device_id']


class ListCall(GroupCall):
    method = _('list')
    rename_fields = {'clientid': 'client_id'}
    int_fields = ['client_id']


class ModuleGraphCall(FileCall):
    method = _('module_graph')
Make module graph call return a file.
Make module graph call return a file.

  
- from ubersmith.calls import BaseCall, GroupCall
+ from ubersmith.calls import BaseCall, GroupCall, FileCall
  from ubersmith.utils import prepend_base
  
  __all__ = [
      'GetCall',
      'ListCall',
+     'ModuleGraphCall',
  ]
  
  _ = prepend_base(__name__.split('.')[-1])
  
  
  class GetCall(BaseCall):
      method = _('get')
      required_fields = ['device_id']
  
  
  class ListCall(GroupCall):
      method = _('list')
      rename_fields = {'clientid': 'client_id'}
      int_fields = ['client_id']
  
  
  class ModuleGraphCall(FileCall):
      method = _('module_graph')
Make module graph call return a file.
## Code Before:

from ubersmith.calls import BaseCall, GroupCall
from ubersmith.utils import prepend_base

__all__ = [
    'GetCall',
    'ListCall',
]

_ = prepend_base(__name__.split('.')[-1])


class GetCall(BaseCall):
    method = _('get')
    required_fields = ['device_id']


class ListCall(GroupCall):
    method = _('list')
    rename_fields = {'clientid': 'client_id'}
    int_fields = ['client_id']


class ModuleGraphCall(FileCall):
    method = _('module_graph')
## Instruction:
Make module graph call return a file.
## Code After:

from ubersmith.calls import BaseCall, GroupCall, FileCall
from ubersmith.utils import prepend_base

__all__ = [
    'GetCall',
    'ListCall',
    'ModuleGraphCall',
]

_ = prepend_base(__name__.split('.')[-1])


class GetCall(BaseCall):
    method = _('get')
    required_fields = ['device_id']


class ListCall(GroupCall):
    method = _('list')
    rename_fields = {'clientid': 'client_id'}
    int_fields = ['client_id']


class ModuleGraphCall(FileCall):
    method = _('module_graph')