from pybricks.hubs import PrimeHub
from pybricks.pupdevices import Motor, ColorSensor, UltrasonicSensor, ForceSensor
from pybricks.parameters import Button, Color, Direction, Port, Side, Stop
from pybricks.robotics import DriveBase
from pybricks.tools import wait, StopWatch
from pybricks.tools import run_task, multitask

hub = PrimeHub()

left_motor = Motor(Port.A, Direction.COUNTERCLOCKWISE)
right_motor = Motor(Port.B)

left_arm = Motor(Port.C, Direction.COUNTERCLOCKWISE)
right_arm = Motor(Port.D)
drive_base = DriveBase(left_motor, right_motor, wheel_diameter=68.8, axle_track=180)
drive_base.settings(600,500,300,200)
drive_base.use_gyro(True)

async def main():
    await right_arm.run_angle(2000,1000)

    await drive_base.straight(200)
    await drive_base.turn(-20)
    await drive_base.straight(525)
    await drive_base.turn(60)
    await drive_base.straight(30)

    await right_arm.run_angle(2000,-1000)
    await drive_base.straight(30)
    await right_arm.run_angle(3000,1000)
    await drive_base.straight(-60)

    await drive_base.turn(-60)
    await drive_base.straight(-525)
    await drive_base.turn(20)
    await drive_base.straight(-200)