Exemplo n.º 1
0
 def test_add_id_2(self):
     schema_in = build_schema('''
         type Human { id: ID! }
     ''')
     config = {'generation': {'field_for_id': True}}
     try:
         generator.run(schema_in, config)
         assert False
     except:
         assert True
Exemplo n.º 2
0
 def test_datetime_scalar_2(self):
     schema_in = build_schema('''
         scalar DateTime
         type datetime_test { date: DateTime! }
     ''')
     config = {'generation': {'generate_datetime': True}}
     try:
         generator.run(schema_in, config)
         assert True
     except:
         assert False
Exemplo n.º 3
0
 def test_add_id_1(self):
     schema_in = build_schema('''
         type Human
     ''')
     config = {'generation': {'field_for_id': True}}
     expected = build_schema('''
         type Human {
             id: ID!
         }
     ''')
     schema_out = generator.run(schema_in, config)
     assert compare.is_equals_schema(schema_out, expected)
Exemplo n.º 4
0
 def test_add_lastUpdateDate_1(self):
     schema_in = build_schema('''
        type Test
     ''')
     config = {'generation': {'field_for_last_update_date': True}}
     expected = build_schema('''
        scalar DateTime
        type Test {
            _lastUpdateDate: DateTime
        }
     ''')
     schema_out = generator.run(schema_in, config)
     assert compare.is_equals_schema(schema_out, expected)
Exemplo n.º 5
0
def creation(self, text_input):
    last_idx = int(sorted(glob.glob('static/results/*'))[-1].split('/')[-1])
    idx = 1 + last_idx
    output_dir = 'static/results/%04d' % idx

    image_path = '{}/{}'.format(output_dir, 'image.jpg')
    config_path = '{}/{}'.format(output_dir, 'config.json')

    config = {
        'text_inputs': [{
            'text': text_input,
            'weight': 10.0
        }],
        'size': (512, 512),
        'num_iterations': 1000,
        'batch_size': 1,
        'cutn': 24,
        'weight_decay': 0.1,
        'learning_rate': 0.1,
        'lr_decay_after': 400,
        'lr_decay_rate': 0.995
    }

    if not os.path.isdir(output_dir):
        os.mkdir(output_dir)

    with open(config_path, 'w') as outfile:
        json.dump(config, outfile)

    def callback(results):
        #status = 'done' if results['iteration'] == results['num_iterations'] else 'generating'
        #status = '%s %d %d'%(status, results['iteration'], results['num_iterations'])
        output = {
            'current': results['iteration'],
            'total': results['num_iterations']
        }
        self.update_state(state='PROGRESS', meta=output)

    img = generator.run(config, callback)
    img.save(image_path)

    results = {
        'current': config['num_iterations'],
        'total': config['num_iterations'],
        'result': image_path
    }

    return results
Exemplo n.º 6
0
 def test_add_input_to_create_1(self):
     schema_in = build_schema('''
         type Human {
             id: ID!
             name: String!
         }
     ''')
     config = {'generation': {'input_to_create_objects': True}}
     expected = build_schema('''
         type Human { 
             id: ID!
             name: String!
         }
         input _InputToCreateHuman {
             name: String!
         }
     ''')
     schema_out = generator.run(schema_in, config)
     assert compare.is_equals_schema(schema_out, expected)
Exemplo n.º 7
0
#!/usr/bin/env python

if __name__=='__main__':
	import generator
	import sys
	generator.run(sys.argv)
Exemplo n.º 8
0
    metavar='str',
    type=str,
    default='',
    help=
    'put this string in the preamble of output file (default: \"%(default)s\")'
)
parser.add_argument('--verbose',
                    action='store_true',
                    help='displays details about the process')
args = parser.parse_args()

with open(args.file, 'r') as file:
    stdout = sys.stdout if args.verbose else open(devnull, 'w')
    print('Read grammar ... ', file=stdout, end='', flush=True)
    cfg = CFG.fromsource(file, args.sep)
    print('ok', file=stdout)
    print('Generate derivation tree ... ', file=stdout, end='', flush=True)
    tree = run(cfg, args.depth, args.size)
    print('ok', file=stdout)
    print('Convert to string ... ', file=stdout, end='', flush=True)
    if args.preamble != '':
        preamble = open(args.preamble, 'r').read() + '\n'
    else:
        preamble = ''
    if args.output == 'stdout':
        print(preamble + tree.to_str(not args.dynamic))
    else:
        with open(args.output, 'w') as output:
            output.write(preamble + tree.to_str())
    print('ok', file=stdout)
Exemplo n.º 9
0
def draw_menu(stdscr):
    response = True
    mode = "main"
    while response:
        if mode == "main":
            stdscr.clear()

            stdscr.addstr(0, 0, "=== SPOTIFY PLAYLIST GENERATOR ===")
            stdscr.addstr(1, 0, "=====       MAIN  MENU       =====")

            stdscr.addstr(2, 2, "[q] = quit")
            stdscr.addstr(3, 2, "[g] = generate playlists")
            stdscr.addstr(4, 2, "[m] = manage categorization")
            # stdscr.addstr(5, 2, "[d] = display")

            msg = "Press a key to select an option:"
            if str(response) != "True":
                msg = response

            stdscr.addstr(7, 0, "> " + msg)

            c = stdscr.getch()

            response = process_menu_input(c)

            if response == "GENERATE_PLAYLIST":
                mode = "generate"
                response = True
            elif response == "MANAGE_CATEGORIES":
                mode = "manage"
                response = True
            elif response == "DISPLAY_DATA":
                mode = "display"
                response = True

        elif mode == "display":
            stdscr.clear()

            stdscr.addstr(0, 0, "=== SPOTIFY PLAYLIST GENERATOR ===")
            stdscr.addstr(1, 0, "=====      DISPLAY DATA      =====")

            stdscr.addstr(2, 2, "[q] = quit")
            stdscr.addstr(3, 2, "[t] = display tracks")
            stdscr.addstr(4, 2, "[a] = display analysis")
            stdscr.addstr(5, 2, "[d] = display definitions")

            msg = "Press a key to select an option:"
            if str(response) != "True":
                msg = response

            stdscr.addstr(7, 0, "> " + msg)

            c = stdscr.getch()

            response = process_display_input(c)

            if response == "DISPLAY_DATA_TRACKS":
                pass
            elif response == "DISPLAY_DATA_ANALYSIS":
                pass
            elif response == "DISPLAY_DATA_DEFINITIONS":
                pass
        elif mode == "generate":
            stdscr.clear()
            stop_curses()
            run()
            mode = "main"
        elif mode == "manage":
            stdscr.clear()
            stop_curses()
            manage()
            mode = "main"

        else:
            break
Exemplo n.º 10
0
def generator_task(generator):
    print "New generator worker started..."
    generator.run()
    return
Exemplo n.º 11
0
import generator as gner

parameters = {
    'ref_type': 'height',
    'ref_h': 60,
    'ref_w': 40,
    'norm_len': 58,
    'resize': 256,
    'min_vail_len': 10,
    'positive_up_boundary': 1.1,
    'positive_low_boundary': 0.9,
    'regression_up_boundary': 1.1,
    'regression_low_boundary': 0.9,
    'ignore_up_boundary': 2,
    'ignore_low_boundary': 0.5,
    'postive_gener_ratio': 0.1,
    'regression_gener_ratio': 0.2,
    'ignore_gener_ratio': 0.4,
    'Image_path': '/mnt/hgfs/linux_share/phone_data/Image',
    'Anno_path': '/mnt/hgfs/linux_share/phone_data/Anno_txt'
}

gner.run(parameters)
Exemplo n.º 12
0
#!/usr/bin/env python

if __name__ == '__main__':
    import generator
    import sys
    generator.run(sys.argv)